learning-loop-node 0.17.0__tar.gz → 0.17.2__tar.gz

{learning_loop_node-0.17.0 → learning_loop_node-0.17.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: learning-loop-node
-Version: 0.17.0
+Version: 0.17.2
 Summary: Python Library for Nodes which connect to the Zauberzeug Learning Loop
 Home-page: https://github.com/zauberzeug/learning_loop_node
 License: MIT

{learning_loop_node-0.17.0 → learning_loop_node-0.17.2}/learning_loop_node/detector/detector_node.py

@@ -522,11 +522,14 @@ class DetectorNode(Node):
         It can be converted e.g. using cv2.imdecode(np.frombuffer(image, np.uint8), cv2.IMREAD_COLOR)"""
 
         await self.detection_lock.acquire()
-        metadata = await run.io_bound(self.detector_logic.evaluate, raw_image)
+        try:
+            metadata = await run.io_bound(self.detector_logic.evaluate, raw_image)
+        finally:
+            self.detection_lock.release()
+
         metadata.tags.extend(tags)
         metadata.source = source
         metadata.created = creation_date
-        self.detection_lock.release()
 
         fix_shape_detections(metadata)
         n_bo, n_cl = len(metadata.box_detections), len(metadata.classification_detections)
@@ -555,8 +558,15 @@ class DetectorNode(Node):
         This function infers the detections from all images, cares about uploading to the loop and returns the detections as a list of ImageMetadata."""
 
         await self.detection_lock.acquire()
-        all_detections = await run.io_bound(self.detector_logic.batch_evaluate, raw_images)
-        self.detection_lock.release()
+        try:
+            all_detections = await run.io_bound(self.detector_logic.batch_evaluate, raw_images)
+        finally:
+            self.detection_lock.release()
+
+        for metadata in all_detections.items:
+            metadata.tags.extend(tags)
+            metadata.source = source
+            metadata.created = creation_date
 
         for detections, raw_image in zip(all_detections.items, raw_images):
             fix_shape_detections(detections)

{learning_loop_node-0.17.0 → learning_loop_node-0.17.2}/learning_loop_node/detector/inbox_filter/cam_observation_history.py

@@ -1,8 +1,14 @@
 import os
 from typing import List, Union
 
-from ...data_classes import (BoxDetection, ClassificationDetection, ImageMetadata, Observation, PointDetection,
-                             SegmentationDetection)
+from ...data_classes import (
+    BoxDetection,
+    ClassificationDetection,
+    ImageMetadata,
+    Observation,
+    PointDetection,
+    SegmentationDetection,
+)
 
 
 class CamObservationHistory:
@@ -10,6 +16,8 @@ class CamObservationHistory:
         self.reset_time = 3600
         self.recent_observations: List[Observation] = []
         self.iou_threshold = 0.5
+        self.min_uncertain_threshold = float(os.environ.get('MIN_UNCERTAIN_THRESHOLD', '0.3'))
+        self.max_uncertain_threshold = float(os.environ.get('MAX_UNCERTAIN_THRESHOLD', '0.6'))
 
     def forget_old_detections(self) -> None:
         self.recent_observations = [detection
@@ -25,7 +33,8 @@ class CamObservationHistory:
                 continue
             if isinstance(detection, ClassificationDetection):
                 # self.recent_observations.append(Observation(detection))
-                causes.add('classification_detection')
+                if self.min_uncertain_threshold <= detection.confidence <= self.max_uncertain_threshold:
+                    causes.add('uncertain')
                 continue
 
             assert isinstance(detection, (BoxDetection, PointDetection)), f"Unknown detection type: {type(detection)}"
@@ -37,7 +46,7 @@ class CamObservationHistory:
                 continue
 
             self.recent_observations.append(Observation(detection))
-            if float(os.environ.get('MIN_UNCERTAIN_THRESHOLD', '0.3')) <= detection.confidence <= float(os.environ.get('MAX_UNCERTAIN_THRESHOLD', '0.6')):
+            if self.min_uncertain_threshold <= detection.confidence <= self.max_uncertain_threshold:
                 causes.add('uncertain')
 
         return list(causes)

{learning_loop_node-0.17.0 → learning_loop_node-0.17.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "learning_loop_node"
-version = "v0.17.0"
+version = "v0.17.2"
 description = "Python Library for Nodes which connect to the Zauberzeug Learning Loop"
 authors = ["Zauberzeug GmbH <info@zauberzeug.com>"]
 license = "MIT"