demos/yolo: list->ndarray (#3593)

Wovchena · web-flow · commit 7fabfbb63a50 · 2022-11-09T14:25:36.000+04:00
Close #3592
diff --git a/demos/common/python/openvino/model_zoo/model_api/models/yolo.py b/demos/common/python/openvino/model_zoo/model_api/models/yolo.py
@@ -477,7 +477,7 @@ def preprocess(self, inputs):
         resized_image = self._change_layout(resized_image)
         dict_inputs = {
             self.image_blob_name: resized_image,
-            self.image_info_blob_name: [[image.shape[0], image.shape[1]]]
+            self.image_info_blob_name: np.array([[image.shape[0], image.shape[1]]], dtype=np.float32)
         }
         return dict_inputs, meta
 

Original file line number	Diff line number	Diff line change
`@@ -477,7 +477,7 @@ def preprocess(self, inputs):`
`477`	`477`	`resized_image = self._change_layout(resized_image)`
`478`	`478`	`dict_inputs = {`
`479`	`479`	`self.image_blob_name: resized_image,`
`480`		`- self.image_info_blob_name: [[image.shape[0], image.shape[1]]]`
	`480`	`+ self.image_info_blob_name: np.array([[image.shape[0], image.shape[1]]], dtype=np.float32)`
`481`	`481`	`}`
`482`	`482`	`return dict_inputs, meta`
`483`	`483`