Update labelImg_2_yolov5.py

Author: SWHL

labelImg_2_yolov5.py

@@ -9,6 +9,7 @@
 
 from tqdm import tqdm
 
+
 class LabelImgToYOLOV5(object):
     def __init__(self, root_dir, out_dir, val_ratio, have_test, test_ratio):
         self.root_dir = Path(root_dir)
@@ -61,7 +62,7 @@ def get_img_list(self):
 
     def gen_image_label_dir(self, img_list):
         new_image_list = []
-        for img_path in img_list:
+        for img_path in tqdm(img_list):
             right_label_path = img_path.with_name(f'{img_path.stem}.txt')
             if right_label_path.exists():
                 self.cp_file(img_path, dst_dir=self.out_img_dir)