读取两个文件夹里不同名的文件，处理映射不对应的文件

# -*- coding: utf-8 -*-
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))  # 差集，在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)

参考文章