python实现原图与掩膜对比

针对增强数据出现的图片数量有出入情况时，通过python实现快速检测出单方面存在的图片

import os


def compare_folders(folder1, folder2):
    folder1_files = set(os.listdir(folder1))
    folder2_files = set(os.listdir(folder2))

    extra_files = folder2_files - folder1_files

    return extra_files


folder1_path = r'path'  # 第一个文件夹的路径
folder2_path = r'path'  # 第二个文件夹的路径


extra_images = compare_folders(folder1_path, folder2_path)

print("多余的图片:")
for image in extra_images:
    print(image)

输出多余文件的名称，结合对比进行删除

def delete_extra_files(folder, files):
    for file in files:
        file_path = os.path.join(folder, file)
        os.remove(file_path)
        print(f"已删除文件: {file_path}")


print("多余的图片:")
for image in extra_images:
    print(image)

delete_extra_files(folder2_path, extra_images)