代码标签不匹配，则删除不匹配项

#代码标签不匹配，则删除不匹配项

import os

img_dir = '/home/imgs'
mask_dir = '/home/masks'

# 获取img和mask文件夹中的文件列表
img_files = set(os.listdir(img_dir))
mask_files = set(os.listdir(mask_dir))

# 找到不匹配的文件
unmatched_img_files = img_files - mask_files
unmatched_mask_files = mask_files - img_files

# 删除不匹配的img文件
for file in unmatched_img_files:
    file_path = os.path.join(img_dir, file)
    if os.path.isfile(file_path):
        os.remove(file_path)
        print(f"Removed unmatched img file: {file_path}")

# 删除不匹配的mask文件
for file in unmatched_mask_files:
    file_path = os.path.join(mask_dir, file)
    if os.path.isfile(file_path):
        os.remove(file_path)
        print(f"Removed unmatched mask file: {file_path}")

print("Done removing unmatched files.")