Align code with polygon-only training

scripts/prepare_polygon_dataset.py

@@ -20,7 +20,7 @@ from PIL import Image
 IMAGE_SUFFIXES = {".jpg", ".jpeg", ".png", ".tif", ".tiff"}
 MASK_SUFFIXES = {".png", ".tif", ".tiff", ".jpg", ".jpeg"}
 POLYGON_FORMATS = {"coco_polygon", "coco_segmentation", "geojson", "shp", "mask", "binary_mask", "semantic_mask"}
-BBOX_FORMATS = {"yolo_bbox_txt", "bbox", "coco_bbox", "voc_bbox"}
+BBOX_FORMATS = {"bbox", "box_txt", "coco_bbox", "voc_bbox"}
 
 
 @dataclass
@@ -39,7 +39,7 @@ class ImportStats:
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser(description=__doc__)
     parser.add_argument("--manifest", required=True, help="JSONL manifest with standardized sample records.")
-    parser.add_argument("--local-yolo-root", default=None, help="Optional local YOLO mirror for bbox rejection auditing.")
+    parser.add_argument("--bbox-source-root", default=None, help="Optional local bbox dataset mirror for rejection auditing.")
     parser.add_argument("--extra-source-root", action="append", default=[], help="Local source roots to scan for mask/polygon datasets.")
     parser.add_argument("--output-root", required=True)
     parser.add_argument("--train-ratio", type=float, default=0.8)
@@ -80,13 +80,13 @@ def local_path_from_record(record: dict[str, Any], key: str) -> Path | None:
     return path if path.exists() else None
 
 
-def find_local_yolo_image(record: dict[str, Any], yolo_root: Path | None) -> Path | None:
-    if yolo_root is None:
+def find_local_bbox_image(record: dict[str, Any], bbox_root: Path | None) -> Path | None:
+    if bbox_root is None:
         return None
     source = str(record.get("image_path") or "")
     stem = Path(source).stem.lower()
     for split in ("train", "val", "test"):
-        image_dir = yolo_root / "images" / split
+        image_dir = bbox_root / "images" / split
         if not image_dir.exists():
             continue
         for path in image_dir.iterdir():
@@ -195,7 +195,7 @@ def main() -> None:
     manifest = Path(args.manifest)
     output_root = Path(args.output_root)
     output_root.mkdir(parents=True, exist_ok=True)
-    yolo_root = Path(args.local_yolo_root) if args.local_yolo_root else None
+    bbox_root = Path(args.bbox_source_root) if args.bbox_source_root else None
     stats = ImportStats()
     accepted: list[dict[str, Any]] = []
     rejected: list[dict[str, Any]] = []
@@ -210,7 +210,7 @@ def main() -> None:
             continue
         quality = float(record.get("quality_score") or 0.0)
         fmt = str(record.get("annotation_format") or "").lower()
-        image_path = local_path_from_record(record, "image_path") or find_local_yolo_image(record, yolo_root)
+        image_path = local_path_from_record(record, "image_path") or find_local_bbox_image(record, bbox_root)
         label_path = local_path_from_record(record, "mask_path") or local_path_from_record(record, "annotation_path")
 
         reject_reason = None