Skip to content

Commit e21fca1

Browse files
committed
Set correct size
1 parent c70800f commit e21fca1

File tree

1 file changed

+1
-1
lines changed
  • demos/common/python/models

1 file changed

+1
-1
lines changed

demos/common/python/models/ssd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def _get_output_parser(self, net, image_blob_name, bboxes='bboxes', labels='labe
6666
pass
6767

6868
try:
69-
parser = BoxesLabelsParser(net.outputs, net.input_info[image_blob_name].input_data.shape[2:])
69+
parser = BoxesLabelsParser(net.outputs, net.input_info[image_blob_name].input_data.shape[2:][::-1])
7070
self.logger.info('Use BoxesLabelsParser')
7171
return parser
7272
except ValueError:

0 commit comments

Comments
 (0)