读取两个文件夹里不同名的文件,处理映射不对应的文件
解决方案:读取两个文件夹里不同名的文件,处理映射不对应的文件
# -*- coding: utf-8 -*-
import ospath1 = 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_namedef 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)) # 差集,在a中但不在b中的元素for name in diff:print("no jpg", name + ".text")diff2 = set(jpg_list).difference(set(text_list)) # 差集,在b中但不在a中的元素print(len(diff2))for name in diff2:print("no txt", name + ".jpg")if __name__ == '__main__':# file_names = read_all_file_name()# print(file_names)file_name(path1, path2)
参考文章