Eduardo_Scripniciuc/st-ten-1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wip

Browse Source
This commit is contained in:
matteo porta 2022-07-27 15:43:03 +02:00
parent e7faf3355f
commit 70c9fb1dbd
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/components/vision.py
View File

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