From 0c4f035728cf474aa566161277e79916432e94df Mon Sep 17 00:00:00 2001
From: Laughing <61612323+Laughing-q@users.noreply.github.com>
Date: Fri, 26 Jan 2024 01:57:00 +0800
Subject: [PATCH] Fix bbox2segment converter (#7814)

---
 ultralytics/data/converter.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ultralytics/data/converter.py b/ultralytics/data/converter.py
index aa3c43e5..0d71b76a 100644
--- a/ultralytics/data/converter.py
+++ b/ultralytics/data/converter.py
@@ -515,6 +515,8 @@ def yolo_bbox2segment(im_dir, save_dir=None, sam_model="sam_b.pt"):
     for l in tqdm(dataset.labels, total=len(dataset.labels), desc="Generating segment labels"):
         h, w = l["shape"]
         boxes = l["bboxes"]
+        if len(boxes) == 0:  # skip empty labels
+            continue
         boxes[:, [0, 2]] *= w
         boxes[:, [1, 3]] *= h
         im = cv2.imread(l["im_file"])