打印data_list

This commit is contained in:
552068321@qq.com 2022-11-07 17:21:45 +08:00
parent f811a51af0
commit c6a8201e4b

View File

@ -179,7 +179,7 @@ def get_file(ori_path: str, type_list: Union[object,str]):
test_files = []
# 训练、测试比例强制91
for img in imgs[0:1]:
path = ori_path + 'images/' +img
path = ori_path + '/images/' +img
# print(os.path.exists(path))
print('图像路径',path)
if os.path.exists(path):
@ -187,18 +187,18 @@ def get_file(ori_path: str, type_list: Union[object,str]):
print('1111')
#label = ori_path + 'labels/' + os.path.split(path)[1]
(filename1, extension) = os.path.splitext(img) # 文件名与后缀名分开
label = ori_path + 'labels/' + filename1 + '.json'
label = ori_path + '/labels/' + filename1 + '.json'
print('标签',label)
if label is not None:
#train_files.append(label)
test_files.append(label)
print('2222')
for img in imgs[1:]:
path = ori_path + 'images/' +img
path = ori_path + '/images/' +img
if os.path.exists(path):
train_files.append(path)
(filename2, extension) = os.path.splitext(img) # 文件名与后缀名分开
label = ori_path + 'labels/' + filename2 + '.json'
label = ori_path + '/labels/' + filename2 + '.json'
if label is not None:
train_files.append(label)