drawbox-cv 0.1.0__py3-none-any.whl

drawbox/__init__.py

@@ -0,0 +1,7 @@
+from .app import AnnotatorApp
+from .config import AnnotatorConfig
+from .dataset import Dataset
+from .box import BoundingBox
+from .classes import ClassRegistry
+
+__all__ = ["AnnotatorApp", "AnnotatorConfig", "Dataset", "BoundingBox", "ClassRegistry"]

drawbox/__main__.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+import sys
+from .app import AnnotatorApp
+from .config import AnnotatorConfig
+
+def main():
+    # Prompt the user for parameters like the original script
+    frames_dir = input("Enter input frames directory path: ").strip()
+    if not frames_dir:
+        print("Error: Input frames directory path is required.")
+        sys.exit(1)
+        
+    output_dir = input("Enter base output directory path (default: output): ").strip()
+    if not output_dir:
+        output_dir = "output"
+
+    start_frame = input("Enter starting frame index (default: 0): ").strip()
+    start_frame = int(start_frame) if start_frame.isdigit() else 0
+    
+    init_lbl = input("Enter initial label file path (optional, press Enter to skip): ").strip()
+    init_fmt = "tf"
+    if init_lbl:
+        init_fmt = input("Enter format of initial label (tf, yolo, coco, voc) (default: tf): ").strip().lower()
+        if not init_fmt:
+            init_fmt = "tf"
+            
+    # Flexible multi-format selection
+    formats_in = input("Enter output format(s) comma-separated (tf, yolo, coco, voc) (default: yolo): ").strip().lower()
+    if not formats_in:
+        formats = ["yolo"]
+    else:
+        formats = [f.strip() for f in formats_in.split(",") if f.strip()]
+    
+    # Validate selected formats
+    valid_formats = {"tf", "yolo", "coco", "voc"}
+    invalid_formats = [f for f in formats if f not in valid_formats]
+    if invalid_formats:
+        print(f"Error: Invalid format(s) selected: {', '.join(invalid_formats)}")
+        print(f"Supported formats are: {', '.join(valid_formats)}")
+        sys.exit(1)
+        
+    carry_over = input("Carry over edited labels to subsequent frames? (y/n) (default: y): ").strip().lower() != "n"
+    save_crops = input("Save cropped bounding boxes? (y/n) (default: n): ").strip().lower() == "y"
+    
+    crops_dir = None
+    if save_crops:
+        crops_dir = input("Enter directory path to save crops (default: output/crops): ").strip()
+        if not crops_dir:
+            crops_dir = None
+            
+    config = AnnotatorConfig(
+        frames_dir=frames_dir,
+        output_dir=output_dir,
+        start_frame=start_frame,
+        initial_label_file=init_lbl,
+        initial_label_format=init_fmt,
+        formats=formats,
+        save_crops=save_crops,
+        crops_out_dir=crops_dir,
+        carry_over_labels=carry_over,
+    )
+    
+    app = AnnotatorApp(config)
+    app.mainloop()
+
+if __name__ == "__main__":
+    main()