Fixed img path error when labelImg to yolov5

RapidAI · Apr 24, 2024 · f16ee51 · f16ee51
1 parent f6c325c
commit f16ee51
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/label_convert/labelImg_to_yolov5.py b/label_convert/labelImg_to_yolov5.py
@@ -90,7 +90,9 @@ def get_train_val_test_list(
         self, img_list, ratio=0.2, have_test=True, test_ratio=0.2
     ):
         random.shuffle(img_list)
-        img_list = [f"{self.out_img_dir / img_path.name}" for img_path in img_list]
+        img_list = [
+            f"{Path(self.out_img_dir.name) / img_path.name}" for img_path in img_list
+        ]
         len_img = len(img_list)
         if have_test:
             split_idx_first = int(len_img * ratio)