|
@@ -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()] # 读取该图片的标签
|