浏览代码

数据加载去除无关打印

liyan 1 年之前
父节点
当前提交
1bae8a063a
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      block/data_get.py

+ 1 - 1
block/data_get.py

@@ -24,7 +24,7 @@ class data_prepare:
         for i in range(len(txt)):
             image_path = f'{self.args.data_path}/image' + txt[i].split('image')[-1]
             data_list[i][0] = image_path
-            print(image_path)
+            # print(image_path)
             lable_path_find = image_path.replace('images', 'labels').replace('.jpg', '.txt')
             with open(lable_path_find, 'r') as f:
                 label_txt = [_.strip().split(' ') for _ in f.readlines()]  # 读取该图片的标签