Browse Source

handle exception when frame isnt in cache

Blake Blackshear 4 năm trước cách đây
mục cha
commit
1785c69e1b
1 tập tin đã thay đổi với 5 bổ sung1 xóa
  1. 5 1
      frigate/object_processing.py

+ 5 - 1
frigate/object_processing.py

@@ -183,7 +183,11 @@ class TrackedObject():
         if self.thumbnail_data is None:
             return None
             
-        best_frame = cv2.cvtColor(self.frame_cache[self.thumbnail_data['frame_time']], cv2.COLOR_YUV2BGR_I420)
+        try:
+            best_frame = cv2.cvtColor(self.frame_cache[self.thumbnail_data['frame_time']], cv2.COLOR_YUV2BGR_I420)
+        except KeyError:
+            logger.warning(f"Unable to create jpg because frame {self.thumbnail_data['frame_time']} is not in the cache")
+            return None
  
         if bounding_box:
             thickness = 2