handle exception when frame isnt in cache

This commit is contained in:
Blake Blackshear 2021-01-19 21:59:35 -06:00
parent e32e69c2d0
commit bb4f79cdfe

View File

@ -183,7 +183,11 @@ class TrackedObject():
if self.thumbnail_data is None: if self.thumbnail_data is None:
return 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: if bounding_box:
thickness = 2 thickness = 2