解决方案:读取两个文件夹里不同名的文件,处理映射不对应的文件
import os
path1 = r'/home/ubuntu/data/yoloData/images/train2017'
path2 = r'/home/ubuntu/data/yoloData/labels/train2017'
def read_all_file_name():
file_path = './text'
file_name = os.listdir(file_path)
return file_name
def file_name(image_dir, text_dir):
jpg_list = []
txt_list = []
jpg_list = os.listdir(image_dir)
text_list = os.listdir(text_dir)
print(len(jpg_list))
print(len(text_list))
for i in range(len(jpg_list)):
jpg_list[i] = jpg_list[i].split(".")[0]
for i in range(len(text_list)):
text_list[i] = text_list[i].split(".")[0]
diff = set(text_list).difference(set(jpg_list))
for name in diff:
print("no jpg", name + ".text")
diff2 = set(jpg_list).difference(set(text_list))
print(len(diff2))
for name in diff2:
print("no txt", name + ".jpg")
if __name__ == '__main__':
file_name(path1, path2)
参考文章