try fix detection handling in upload

zauberzeug · Dec 6, 2023 · ee76b40 · ee76b40
1 parent 173bb1c
commit ee76b40
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/learning_loop_node/detector/detector_logic.py b/learning_loop_node/detector/detector_logic.py
@@ -1,6 +1,6 @@
 import logging
 from abc import abstractmethod
-from typing import Any, Optional
+from typing import Optional
 
 import numpy as np
 

diff --git a/learning_loop_node/detector/detector_node.py b/learning_loop_node/detector/detector_node.py
@@ -145,9 +145,14 @@ async def _upload(sid, data: Dict) -> Optional[Dict]:
                 detections = from_dict(data_class=Detections, data=detection_data)
                 detections = self.add_category_id_to_detections(self.detector_logic.model_info, detections)
             else:
+                print(f'No detections: {detection_data}', flush=True)
                 detections = Detections()
 
-            tags = data.get('tags', ['picked_by_system'])
+            tags = data.get('tags', [])
+            tags.append('picked_by_system')
+            camera_id = data.get('camera-id', None) or data.get('mac', None)
+            if camera_id is not None:
+                tags.append(camera_id)
 
             loop = asyncio.get_event_loop()
             try: