Merge pull request openvinotoolkit#2194 from eizamaliev/py_detection_…

…demo_fix Fix Python's object detection demo for support person-detection-0203 model
aDanPin · Mar 12, 2021 · 26d6aba · 26d6aba
2 parents c70800f + e21fca1
commit 26d6aba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demos/common/python/models/ssd.py b/demos/common/python/models/ssd.py
@@ -66,7 +66,7 @@ def _get_output_parser(self, net, image_blob_name, bboxes='bboxes', labels='labe
             pass
 
         try:
-            parser = BoxesLabelsParser(net.outputs, net.input_info[image_blob_name].input_data.shape[2:])
+            parser = BoxesLabelsParser(net.outputs, net.input_info[image_blob_name].input_data.shape[2:][::-1])
             self.logger.info('Use BoxesLabelsParser')
             return parser
         except ValueError: