Update

tests/show_yolo_seg.py

@@ -56,9 +56,7 @@ def yolo_bbox_to_xyxy(coords, img_w, img_h):
 def poly_to_pts(coords, img_w, img_h):
     # coords: [x1 y1 x2 y2 ...] either normalized or absolute
     if coords_are_normalized(coords[4:]):
-        coords = [
-            coords[i] * (img_w if i % 2 == 0 else img_h) for i in range(len(coords))
-        ]
+        coords = [coords[i] * (img_w if i % 2 == 0 else img_h) for i in range(len(coords))]
     pts = np.array(coords, dtype=np.int32).reshape(-1, 2)
     return pts
 
@@ -154,21 +152,21 @@ def load_labels_file(label_path):
 
 
 def main():
-    parser = argparse.ArgumentParser(
-        description="Show YOLO segmentation / polygon annotations"
-    )
+    parser = argparse.ArgumentParser(description="Show YOLO segmentation / polygon annotations")
     parser.add_argument("image", type=str, help="Path to image file")
-    parser.add_argument("labels", type=str, help="Path to YOLO label file (polygons)")
-    parser.add_argument(
-        "--alpha", type=float, default=0.4, help="Polygon fill alpha (0..1)"
-    )
-    parser.add_argument(
-        "--no-bbox", action="store_true", help="Don't draw bounding boxes for polygons"
-    )
+    parser.add_argument("--labels", type=str, help="Path to YOLO label file (polygons)")
+    parser.add_argument("--alpha", type=float, default=0.4, help="Polygon fill alpha (0..1)")
+    parser.add_argument("--no-bbox", action="store_true", help="Don't draw bounding boxes for polygons")
     args = parser.parse_args()
 
+    print(args)
+
     img_path = Path(args.image)
-    lbl_path = Path(args.labels)
+    if args.labels:
+        lbl_path = Path(args.labels)
+    else:
+        lbl_path = img_path.with_suffix(".txt")
+        lbl_path = Path(str(lbl_path).replace("images", "labels"))
 
     if not img_path.exists():
         print("Image not found:", img_path)
@@ -188,9 +186,7 @@ def main():
     if not labels:
         print("No labels parsed from", lbl_path)
         # continue and just show image
-    out = draw_annotations(
-        img.copy(), labels, alpha=args.alpha, draw_bbox_for_poly=(not args.no_bbox)
-    )
+    out = draw_annotations(img.copy(), labels, alpha=args.alpha, draw_bbox_for_poly=(not args.no_bbox))
 
     # Convert BGR -> RGB for matplotlib display
     out_rgb = cv2.cvtColor(out, cv2.COLOR_BGR2RGB)