tensorflow-lite: fix prediction crash

koush · Apr 8, 2024 · 10a7877 · 10a7877
1 parent f15526f
commit 10a7877
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json
diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json
@@ -53,5 +53,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.51"
+   "version": "0.1.52"
 }
diff --git a/plugins/tensorflow-lite/src/predict/__init__.py b/plugins/tensorflow-lite/src/predict/__init__.py
@@ -122,7 +122,7 @@ def create_detection_result(self, objs: List[Prediction], size, convert_to_src_s
                 obj.bbox.xmin, obj.bbox.ymin, obj.bbox.xmax - obj.bbox.xmin, obj.bbox.ymax - obj.bbox.ymin)
             detection['className'] = className
             detection['score'] = obj.score
-            if obj.embedding is not None:
+            if hasattr(obj, 'embedding') and obj.embedding is not None:
                 detection['embedding'] = obj.embedding
             detections.append(detection)