wip

src/components/vision.py

@@ -336,6 +336,8 @@ class Vision(Component):
         self.loading_model_signal.emit({"status": "done"})
 
     def check_features(self, frame, lock=True):
+        if frame is None:
+            return []
         if self.interpreter is not None and frame.shape != self.interpreter.get_input_details()[0]["shape"][1:3]:
             tensor = np.expand_dims(cv2.resize(frame, self.interpreter.get_input_details()[0]["shape"][1:3], interpolation=cv2.INTER_LINEAR), axis=0)
         else:
@@ -691,7 +693,7 @@ class Vision(Component):
 
     def vision_consumer_work(self, consumable=None):
         # VISION_CONSUMER TASK
-        if consumable is None:
+        if consumable is None or consumable["frame"] is None:
             return
         detections = self.check_features(consumable["frame"])
         results = self.process_detections(detections)