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

argus/__init__.py

@@ -1,3 +1,3 @@
 """Argus - Vision AI dataset toolkit."""
 
-__version__ = "1.3.0"
+__version__ = "1.4.0"

argus/cli.py

@@ -534,13 +534,6 @@ def split_dataset(
             help="Train/val/test ratio (e.g. 0.8,0.1,0.1).",
         ),
     ] = "0.8,0.1,0.1",
-    stratify: Annotated[
-        bool,
-        typer.Option(
-            "--stratify/--no-stratify",
-            help="Stratify by class distribution when splitting.",
-        ),
-    ] = True,
     seed: Annotated[
         int,
         typer.Option(
@@ -593,9 +586,7 @@ def split_dataset(
         ) as progress:
             progress.add_task("Creating YOLO splits...", total=None)
             try:
-                counts = split_yolo_dataset(
-                    dataset, output_path, ratios, stratify, seed
-                )
+                counts = split_yolo_dataset(dataset, output_path, ratios, True, seed)
             except ValueError as exc:
                 console.print(f"[red]Error: {exc}[/red]")
                 raise typer.Exit(1) from exc
@@ -628,7 +619,7 @@ def split_dataset(
                     annotation_file,
                     output_path,
                     ratios,
-                    stratify,
+                    True,
                     seed,
                 )
             except ValueError as exc:
@@ -1352,9 +1343,12 @@ def _draw_annotations(
             overlay = img.copy()
             cv2.fillPoly(overlay, [pts], color)
             cv2.addWeighted(overlay, 0.3, img, 0.7, 0, img)
+            # Draw small points at polygon vertices
+            for pt in pts:
+                cv2.circle(img, tuple(pt), radius=3, color=color, thickness=-1)
 
-        # Draw bounding box
-        if bbox:
+        # Draw bounding box (only for detection, not segmentation)
+        if bbox and not polygon:
             x, y, w, h = bbox
             x1, y1 = int(x), int(y)
             x2, y2 = int(x + w), int(y + h)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: argus-cv
-Version: 1.3.0
+Version: 1.4.0
 Summary: CLI tool for working with vision AI datasets
 Requires-Python: >=3.10
 Requires-Dist: numpy>=1.24.0
@@ -12,7 +12,7 @@ Description-Content-Type: text/markdown
 
 # argus-cv
 
-Vision AI dataset toolkit for working with YOLO and COCO datasets.
+Vision AI dataset toolkit for working with YOLO, COCO, and semantic mask datasets.
 
 **[Documentation](https://pirnerjonas.github.io/argus/)**
 
@@ -22,6 +22,13 @@ Vision AI dataset toolkit for working with YOLO and COCO datasets.
 uvx argus-cv
 ```
 
+## Highlights
+
+- Detects YOLO, COCO, and folder-based semantic mask datasets.
+- Reports per-class stats (pixel coverage for mask datasets).
+- Interactive viewer with bounding boxes, polygons, or mask overlays.
+- Split unsplit YOLO/COCO datasets into train/val/test.
+
 ## Usage
 
 ```bash

{argus_cv-1.3.0.dist-info → argus_cv-1.4.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
-argus/__init__.py,sha256=277ASQvH6ZWVWUzOCVB7vAxn3OYciow4nVkjG16-kio,64
+argus/__init__.py,sha256=rRRcnsvIBvQofIEjwsDsXsT9TiGB9c6zVOmzqmqX4Qg,64
 argus/__main__.py,sha256=63ezHx8eL_lCMoZrCbKhmpao0fmdvYVw1chbknGg-oI,104
-argus/cli.py,sha256=th1Rgn1Sm9juWoavopEXLBT8XEh5lKzOMX-pccwvDgA,47904
+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
@@ -8,7 +8,7 @@ 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.3.0.dist-info/METADATA,sha256=1CCYLSnGHaAdS5jjwUuTJWRHu5OM_oFfvi1L_v3SkQw,1070
-argus_cv-1.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-argus_cv-1.3.0.dist-info/entry_points.txt,sha256=dvJFH7BkrOxJnifSjPhwq1YCafPaqdngWyBuFYE73yY,43
-argus_cv-1.3.0.dist-info/RECORD,,
+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,,