argus-cv 1.4.0__py3-none-any.whl → 1.5.1__py3-none-any.whl

argus/core/yolo.py

@@ -1,5 +1,6 @@
 """YOLO dataset detection and handling."""
 
+import json
 from dataclasses import dataclass, field
 from pathlib import Path
 
@@ -200,6 +201,12 @@ class YOLODataset(Dataset):
         if not class_dirs:
             return None
 
+        # Check if any directory contains COCO annotation files
+        # (Roboflow COCO datasets have train/valid/test dirs with JSON files)
+        for class_dir in class_dirs:
+            if cls._has_coco_annotation(class_dir):
+                return None  # Not a classification dataset
+
         # Check if these are class directories (contain images directly)
         class_names_set = set()
         for class_dir in class_dirs:
@@ -225,6 +232,28 @@ class YOLODataset(Dataset):
             config_file=None,
         )
 
+    @staticmethod
+    def _has_coco_annotation(directory: Path) -> bool:
+        """Check if directory contains COCO annotation files.
+
+        Args:
+            directory: Directory to check.
+
+        Returns:
+            True if COCO annotation files are found, False otherwise.
+        """
+        for f in directory.glob("*.json"):
+            try:
+                with open(f, encoding="utf-8") as fp:
+                    data = json.load(fp)
+                if isinstance(data, dict) and any(
+                    k in data for k in ["images", "annotations", "categories"]
+                ):
+                    return True
+            except (json.JSONDecodeError, OSError):
+                pass
+        return False
+
     def get_instance_counts(self) -> dict[str, dict[str, int]]:
         """Get the number of annotation instances per class, per split.
 

{argus_cv-1.4.0.dist-info → argus_cv-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: argus-cv
-Version: 1.4.0
+Version: 1.5.1
 Summary: CLI tool for working with vision AI datasets
 Requires-Python: >=3.10
 Requires-Dist: numpy>=1.24.0

argus_cv-1.5.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+argus/__init__.py,sha256=Yo7UDKujDodxid1b2g022IqmD1bwc9POtFSl8iolq5c,64
+argus/__main__.py,sha256=63ezHx8eL_lCMoZrCbKhmpao0fmdvYVw1chbknGg-oI,104
+argus/cli.py,sha256=Xri1KFVOMS-YhbwkRE0eB5HOf49kfQuftu0IJU4gAjA,59605
+argus/commands/__init__.py,sha256=i2oor9hpVpF-_1qZWCGDLwwi1pZGJfZnUKJZ_NMBG18,30
+argus/core/__init__.py,sha256=L5Onny8UjJtok5hOBKftqwnOhPvgQS5MZT_kgLpes1o,928
+argus/core/base.py,sha256=WBrB7XWz125YZ1UQfHQwsYAuIFY_XGEhG_0ybgPhn6s,3696
+argus/core/coco.py,sha256=V3Ifh6KUbifBTLefUuMxQkejgkwsPZNfKLn0newDZJ4,17539
+argus/core/convert.py,sha256=cHuw1E9B4vyozpikS2PJnFfiJ_eRMPIHblizyeZz1Ps,8471
+argus/core/filter.py,sha256=7BRefzYcKIxU0GkFNHiJJAijc9UIhrvNKdYgXE_22ig,21945
+argus/core/mask.py,sha256=m7Ztf4lAZx5ITpk3F3mETcvCC6hGydlxK0-2nCjeTfU,21835
+argus/core/split.py,sha256=kEWtbdg6bH-WiNFf83HkqZD90EL4gsavw6JiefuAETs,10776
+argus/core/yolo.py,sha256=Vtw2sga40VooaRE8bmjKtr_aYhfoV7ZcVijFjg1DVwo,29644
+argus_cv-1.5.1.dist-info/METADATA,sha256=71pcUGzx6s0uCiJjZNUh-4p-z50xCL5xqQC8-JXhNaI,1353
+argus_cv-1.5.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+argus_cv-1.5.1.dist-info/entry_points.txt,sha256=dvJFH7BkrOxJnifSjPhwq1YCafPaqdngWyBuFYE73yY,43
+argus_cv-1.5.1.dist-info/RECORD,,

argus_cv-1.4.0.dist-info/RECORD

@@ -1,14 +0,0 @@
-argus/__init__.py,sha256=rRRcnsvIBvQofIEjwsDsXsT9TiGB9c6zVOmzqmqX4Qg,64
-argus/__main__.py,sha256=63ezHx8eL_lCMoZrCbKhmpao0fmdvYVw1chbknGg-oI,104
-argus/cli.py,sha256=CXnzBiD0ROjHOVac1Jd3Yre4jPeiqkb1F0TGjHyccwc,47879
-argus/commands/__init__.py,sha256=i2oor9hpVpF-_1qZWCGDLwwi1pZGJfZnUKJZ_NMBG18,30
-argus/core/__init__.py,sha256=II2wYJpGUUGGKOFZ5BCpMIBTfv0WP-F15U_xbpWGjk8,453
-argus/core/base.py,sha256=WBrB7XWz125YZ1UQfHQwsYAuIFY_XGEhG_0ybgPhn6s,3696
-argus/core/coco.py,sha256=atVurZV2T7cszydyD9GfDTWHGYDd-JNK5RD7nse5avc,15823
-argus/core/mask.py,sha256=m7Ztf4lAZx5ITpk3F3mETcvCC6hGydlxK0-2nCjeTfU,21835
-argus/core/split.py,sha256=kEWtbdg6bH-WiNFf83HkqZD90EL4gsavw6JiefuAETs,10776
-argus/core/yolo.py,sha256=tTc9jJzXcwa8LQ_s8nv-D_i2b9k_x-LT1O0eWr4sZ2k,28616
-argus_cv-1.4.0.dist-info/METADATA,sha256=Gdf_n7cxFqS2JoM2Z48CqFcFWMAYHT1y3X1iy9WAxTA,1353
-argus_cv-1.4.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-argus_cv-1.4.0.dist-info/entry_points.txt,sha256=dvJFH7BkrOxJnifSjPhwq1YCafPaqdngWyBuFYE73yY,43
-argus_cv-1.4.0.dist-info/RECORD,,