Skip to content

Instantly share code, notes, and snippets.

@rmccorm4
Created May 10, 2020 21:44
Show Gist options
  • Save rmccorm4/415c39413dfa5d4c8908ad2377d36e91 to your computer and use it in GitHub Desktop.
Save rmccorm4/415c39413dfa5d4c8908ad2377d36e91 to your computer and use it in GitHub Desktop.
{
'Profile 0':
{
'actual_input_1':
{
'binding_dtype': DataType.FLOAT,
'binding_index': 0,
'binding_name': 'actual_input_1', # context.get_binding_name(binding_index)
'binding_shape': (-1, 3, 224, 224), # context.get_binding_shape(binding_index)
'binding_type': 'INPUT', # engine.binding_is_input(binding_index) == True
'profile': 0,
'profile_shape': [(1, 3, 224, 224), # min
(32, 3, 224, 224), # opt
(32, 3, 224, 224)] # max
},
'output1':
{
'binding_dtype': DataType.FLOAT,
'binding_index': 1,
'binding_name': 'output1', # context.get_binding_name(binding_index)
'binding_shape': (-1, 1000), # context.get_binding_shape(binding_index)
'binding_type': 'OUTPUT', # engine.binding_is_input(binding_index) == False
'profile': 0
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment