openvino: avoid ov.Tensor when using start_async due to thread safety?

koush · Dec 31, 2024 · a41e915 · a41e915
1 parent f0db59f
commit a41e915
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 9 deletions.
diff --git a/plugins/openvino/package-lock.json b/plugins/openvino/package-lock.json
diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json
@@ -48,5 +48,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.149"
+   "version": "0.1.150"
 }
diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py
@@ -375,14 +375,11 @@ def prepare():
                         im = im.reshape((1, 3, self.model_dim, self.model_dim))
                 im = im.astype(np.float32) / 255.0
                 im = np.ascontiguousarray(im)  # contiguous
-                input_tensor = ov.Tensor(array=im)
             elif self.yolo:
-                input_tensor = ov.Tensor(
-                    array=np.expand_dims(np.array(input), axis=0).astype(np.float32)
-                )
+                im = np.expand_dims(np.array(input), axis=0).astype(np.float32)
             else:
-                input_tensor = ov.Tensor(array=np.expand_dims(np.array(input), axis=0))
-            return input_tensor
+                im = np.expand_dims(np.array(input), axis=0)
+            return im
 
         try:
             input_tensor = await asyncio.get_event_loop().run_in_executor(