EasyMetrics 0.1.0__py3-none-any.whl → 0.1.5__py3-none-any.whl

easyMetrics/tasks/detection/format_converter.py

@@ -84,8 +84,18 @@ class DetectionFormatConverter:
     def _convert_coco_preds(coco_preds: List[Any]) -> List[Dict[str, Any]]:
         """
         转换 COCO 格式的预测结果。
-        COCO 格式示例: [{"image_id": 0, "category_id": 0, "bbox": [0, 0, 100, 100], "score": 0.9}]
+        
+        支持两种格式：
+        1. COCO 官方格式: [{"image_id": 0, "category_id": 0, "bbox": [0, 0, 100, 100], "score": 0.9}]
+        2. 内部统一格式: [{"boxes": [[x1, y1, x2, y2]], "scores": [0.9], "labels": [0]}]
         """
+        # 检查是否已经是内部统一格式
+        if coco_preds and isinstance(coco_preds[0], dict):
+            first_pred = coco_preds[0]
+            if 'boxes' in first_pred and 'scores' in first_pred and 'labels' in first_pred:
+                # 已经是内部统一格式，直接返回
+                return coco_preds
+        
         # 按 image_id 分组
         grouped_preds = {}
         for pred in coco_preds:
@@ -111,8 +121,18 @@ class DetectionFormatConverter:
     def _convert_coco_targets(coco_targets: List[Any]) -> List[Dict[str, Any]]:
         """
         转换 COCO 格式的真实标签。
-        COCO 格式示例: [{"image_id": 0, "category_id": 0, "bbox": [0, 0, 100, 100], "area": 10000, "iscrowd": 0}]
+        
+        支持两种格式：
+        1. COCO 官方格式: [{"image_id": 0, "category_id": 0, "bbox": [0, 0, 100, 100], "area": 10000, "iscrowd": 0}]
+        2. 内部统一格式: [{"boxes": [[x1, y1, x2, y2]], "labels": [0]}]
         """
+        # 检查是否已经是内部统一格式
+        if coco_targets and isinstance(coco_targets[0], dict):
+            first_target = coco_targets[0]
+            if 'boxes' in first_target and 'labels' in first_target:
+                # 已经是内部统一格式，直接返回
+                return coco_targets
+        
         # 按 image_id 分组
         grouped_targets = {}
         for target in coco_targets:

{easymetrics-0.1.0.dist-info → easymetrics-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: EasyMetrics
-Version: 0.1.0
+Version: 0.1.5
 Summary: 轻量级、零依赖的机器学习指标评估平台
 Author-email: EasyMetrics Team <team@easymetrics.example>
 License: MIT License

{easymetrics-0.1.0.dist-info → easymetrics-0.1.5.dist-info}/RECORD

@@ -3,13 +3,13 @@ easyMetrics/core/__init__.py,sha256=L-JsvnYEbmJZPzf8oBbldRA-B6Oni3AtJaZZJEuDOwA,
 easyMetrics/core/base.py,sha256=pMRZCl_SJtfCo-_fzuGijkYMSq1DDIZegI96EWxUX6k,789
 easyMetrics/tasks/__init__.py,sha256=Cwc3WM7ZgblWq1jfIjrHt_bBrJRobx4q_WbHn3LaN68,44
 easyMetrics/tasks/detection/__init__.py,sha256=74SsA-fJbamX7WGIWJJmeTXjjHnogYU1jwYse65d45U,80
-easyMetrics/tasks/detection/format_converter.py,sha256=EhGz-8j46bf5higaKyJmtqwILp8V2NtpiK0i-gOAgdE,11700
+easyMetrics/tasks/detection/format_converter.py,sha256=80vM8B3exuH6EHlmYwvMrpslTDm-FMWzuccQKka42dc,12632
 easyMetrics/tasks/detection/interface.py,sha256=ZbR9aJVXsRiSNGrhcVZAEGkfH2SFX0kJi0XY_rUEbMA,2568
 easyMetrics/tasks/detection/map.py,sha256=F2qVDTQR1qr1SHEf3w3Xg7yvUytzfQtxXdF6p25CcwA,17471
 easyMetrics/tasks/detection/matcher.py,sha256=C-4hEVBJseQKgcHupBLlaDNFGhKxOxN-YdPNnpuit9I,2443
 easyMetrics/tasks/detection/utils.py,sha256=jNDPHxWzGo7478QDI1f8uVGK6HUlwvHlPbfwzEk-H0A,2770
-easymetrics-0.1.0.dist-info/licenses/LICENSE,sha256=Z-thSEoGCfxHrpC8rpgrqY-LSoJuyHbgPj5Gt8BuG_0,1073
-easymetrics-0.1.0.dist-info/METADATA,sha256=dHEOCgWpY4ECOtByCGk-0fJYz4lFYSq18oYnMCte4Eg,5265
-easymetrics-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-easymetrics-0.1.0.dist-info/top_level.txt,sha256=2ytZxZ9LMTdKpeZunCE1rpvkpDAhWeQLiiXTrFy9niw,12
-easymetrics-0.1.0.dist-info/RECORD,,
+easymetrics-0.1.5.dist-info/licenses/LICENSE,sha256=Z-thSEoGCfxHrpC8rpgrqY-LSoJuyHbgPj5Gt8BuG_0,1073
+easymetrics-0.1.5.dist-info/METADATA,sha256=2iaYmKt03dFhBsxudv6nFfiOLrmQ23Vsqm9mIwaCqMs,5265
+easymetrics-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+easymetrics-0.1.5.dist-info/top_level.txt,sha256=2ytZxZ9LMTdKpeZunCE1rpvkpDAhWeQLiiXTrFy9niw,12
+easymetrics-0.1.5.dist-info/RECORD,,