xkrs_manan/RODY
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

打印data_list

Browse Source
This commit is contained in:
552068321@qq.com 2022-11-07 17:21:45 +08:00
parent f811a51af0
commit c6a8201e4b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/file_tool.py
View File

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