同名掩膜图片，删除不共有文件

import os
# 指定文件夹路径
data_folder="./"
folder_path = data_folder+"masks"

# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    # 检查文件是否以 m.png 结尾
    if filename.lower().endswith("m.png"):
    # if filename.lower().endswith(".jpg"):
        # 构建新的文件名，替换为.png
        new_filename = filename[:-5] + ".png"
        # new_filename = filename[:-4] + ".png"
        # 构建完整的文件路径
        old_file_path = os.path.join(folder_path, filename)
        new_file_path = os.path.join(folder_path, new_filename)
        # 重命名文件
        os.rename(old_file_path, new_file_path)
        print(f" {old_file_path} to {new_file_path}")
# In[]
#删除不共有的元素
import os
images_folder = 'images'
labels_folder = 'masks'
# 获取images文件夹中的所有文件名（不包括后缀）
image_files = [os.path.splitext(file)[0] for file in os.listdir(images_folder)]
# 获取labels文件夹中的所有文件名（不包括后缀）
label_files = [os.path.splitext(file)[0] for file in os.listdir(labels_folder)]
set1 = set(image_files)
set2 = set(label_files)

# 在list1中有，但在list2中没有的元素
elements_only_in_list1 = list(set1 - set2)

# 在list2中有，但在list1中没有的元素
elements_only_in_list2 = list(set2 - set1)

print("list1中有，但list2中没有的元素:", elements_only_in_list1)
print("list2中有，但list1中没有的元素:", elements_only_in_list2)

# 指定目录路径
directory = 'images/'
for item in elements_only_in_list1 :
    print("item",item)
    file_path=directory+item+".png"
    print("file_path", file_path)
    os.remove(file_path)

# %%