property to get latest image which has detections

zauberzeug · Nov 13, 2023 · 617cc17 · 617cc17
1 parent 9f2042d
commit 617cc17
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rosys/vision/camera.py b/rosys/vision/camera.py
@@ -34,6 +34,10 @@ def latest_captured_image(self) -> Optional[Image]:
         images = self.captured_images
         return images[-1] if images else None
 
+    @property
+    def latest_detected_image(self) -> Optional[Image]:
+        return next((i for i in self.captured_images if i.detections), None)
+
     def get_recent_images(self, current_time: float, timespan: float = 10.0) -> list[Image]:
         return [i for i in self.captured_images if i.time > current_time - timespan]