matrice-analytics 0.1.3__py3-none-any.whl → 0.1.31__py3-none-any.whl

matrice_analytics/post_processing/usecases/flare_analysis.py

@@ -13,7 +13,6 @@ from ..utils import (
     filter_by_categories, 
     apply_category_mapping, 
     match_results_structure,
-    extract_major_colors,
     bbox_smoothing,
     BBoxSmoothingConfig,
     BBoxSmoothingTracker
@@ -462,60 +461,6 @@ class FlareAnalysisUseCase(BaseProcessor):
         is_video = self._is_video_bytes(media_bytes)
         return self._analyze_flares_in_video(data, media_bytes, config) if is_video else self._analyze_flares_in_image(data, media_bytes, config)
 
-    def _analyze_flares_in_video(self, data: Any, video_bytes: bytes, config: FlareAnalysisConfig) -> List[Dict[str, Any]]:
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_video:
-            temp_video.write(video_bytes)
-            video_path = temp_video.name
-        try:
-            cap = cv2.VideoCapture(video_path)
-            if not cap.isOpened():
-                raise RuntimeError("Failed to open video file")
-            fps = config.fps or cap.get(cv2.CAP_PROP_FPS)
-            flare_analysis = []
-            frame_id = 0
-            while True:
-                ret, frame = cap.read()
-                if not ret:
-                    break
-                if frame_id % config.frame_skip != 0:
-                    frame_id += 1
-                    continue
-                frame_key = str(frame_id)
-                timestamp = frame_id / fps
-                frame_detections = self._get_frame_detections(data, frame_key)
-                if not frame_detections:
-                    frame_id += 1
-                    continue
-                rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-                for detection in frame_detections:
-                    if detection.get("confidence", 1.0) < config.confidence_threshold:
-                        continue
-                    bbox = detection.get("bounding_box", detection.get("bbox"))
-                    if not bbox:
-                        continue
-                    crop = self._crop_bbox(rgb_frame, bbox, config.bbox_format)
-                    if crop.size == 0:
-                        continue
-                    major_colors = extract_major_colors(crop, k=config.top_k_colors)
-                    main_color = major_colors[0][0] if major_colors else "unknown"
-                    flare_record = {
-                        "frame_id": frame_key,
-                        "timestamp": round(timestamp, 2),
-                        "category": detection.get("category", "unknown"),
-                        "confidence": round(detection.get("confidence", 0.0), 3),
-                        "main_color": main_color,
-                        "major_colors": major_colors,
-                        "bounding_box": bbox,
-                        "detection_id": detection.get("id", f"det_{len(flare_analysis)}"),
-                        "track_id": detection.get("track_id")
-                    }
-                    flare_analysis.append(flare_record)
-                frame_id += 1
-            cap.release()
-            return flare_analysis
-        finally:
-            if os.path.exists(video_path):
-                os.unlink(video_path)
 
     def _analyze_flares_in_image(self, data: Any, image_bytes: bytes, config: FlareAnalysisConfig) -> List[Dict[str, Any]]:
         image_array = np.frombuffer(image_bytes, np.uint8)
@@ -534,7 +479,7 @@ class FlareAnalysisUseCase(BaseProcessor):
             crop = self._crop_bbox(rgb_image, bbox, config.bbox_format)
             if crop.size == 0:
                 continue
-            major_colors = extract_major_colors(crop, k=config.top_k_colors)
+            major_colors = [()] #extract_major_colors(crop, k=config.top_k_colors)
             main_color = major_colors[0][0] if major_colors else "unknown"
             flare_record = {
                 "frame_id": "0",