matrice 1.0.99303__tar.gz → 1.0.99304__tar.gz

{matrice-1.0.99303/src/matrice.egg-info → matrice-1.0.99304}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99303
+Version: 1.0.99304
 Summary: SDK for connecting to matrice.ai services
 Home-page: https://github.com/matrice-ai/python-sdk
 Author: Matrice.ai

{matrice-1.0.99303 → matrice-1.0.99304}/setup.py

@@ -14,7 +14,7 @@ long_description = (here / "README.md").read_text(encoding="utf-8")
 setup(
     name="matrice",
 
-    version = "1.0.99303",
+    version = "1.0.99304",
     
     description="SDK for connecting to matrice.ai services",
     long_description=long_description,

{matrice-1.0.99303 → matrice-1.0.99304}/src/matrice/deploy/utils/post_processing/usecases/color_detection.py

@@ -261,8 +261,9 @@ class ColorDetectionUseCase(BaseProcessor):
             cat = det.get('category')
             color = det.get('main_color')
             track_id = det.get('track_id')
-            if cat and color and track_id is not None:
-                key = f"{cat}:{color}"  # Unique key for category-color pair
+            if cat and track_id is not None:
+                # If color not yet computed at this stage, fall back to category-only key
+                key = f"{cat}:{color}" if color else cat
                 self._color_total_track_ids[key].add(track_id)
                 self._color_current_frame_track_ids[key].add(track_id)
 
@@ -419,10 +420,22 @@ class ColorDetectionUseCase(BaseProcessor):
             print("-------------------COLOR_ANALYSIS-------------------")
             
             # Step 8: Calculate summaries
+            # After color extraction, update cumulative color-aware tracking totals
+            self._update_color_tracking_state_from_analysis(color_analysis)
             color_summary = self._calculate_color_summary(color_analysis, config)
+            # Ensure total_color_counts is populated even on first frame/session
+            totals = self.get_total_color_counts()
+            if not totals:
+                tmp = defaultdict(set)
+                for rec in color_analysis:
+                    cat = rec.get('category')
+                    tid = rec.get('track_id') or rec.get('detection_id')
+                    if cat and tid is not None:
+                        tmp[cat].add(tid)
+                totals = {cat: len(ids) for cat, ids in tmp.items()}
             
             general_summary = self._calculate_general_summary(processed_data, config)
-            color_summary['total_color_counts'] = self.get_total_color_counts()
+            color_summary['total_color_counts'] = totals
             print("-------------------COLOR_SUMMARY-------------------")
             print(color_summary)
             print("-------------------COLOR_SUMMARY-------------------")
@@ -511,6 +524,25 @@ class ColorDetectionUseCase(BaseProcessor):
             return self._analyze_colors_in_video(data, media_bytes, config)
         else:
             return self._analyze_colors_in_image(data, media_bytes, config)
+
+    def _update_color_tracking_state_from_analysis(self, color_analysis: List[Dict[str, Any]]) -> None:
+        """Update total tracking store using analyzed color results.
+        Ensures totals are populated even if pre-analysis detections lacked colors/track_ids."""
+        existing_store = getattr(self, '_color_total_track_ids', None)
+        if not isinstance(existing_store, defaultdict):
+            existing_store = {} if existing_store is None else dict(existing_store)
+            self._color_total_track_ids = defaultdict(set, existing_store)
+        else:
+            self._color_total_track_ids = existing_store
+        for rec in color_analysis:
+            cat = rec.get('category')
+            color = rec.get('main_color')
+            track_id = rec.get('track_id')
+            if track_id is None:
+                track_id = rec.get('detection_id')
+            if cat and track_id is not None:
+                key = f"{cat}:{color}" if color else cat
+                self._color_total_track_ids[key].add(track_id)
     
     def _is_video_bytes(self, media_bytes: bytes) -> bool:
         """Determine if bytes represent a video file."""

{matrice-1.0.99303 → matrice-1.0.99304/src/matrice.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99303
+Version: 1.0.99304
 Summary: SDK for connecting to matrice.ai services
 Home-page: https://github.com/matrice-ai/python-sdk
 Author: Matrice.ai