import os

# 定义两个目录的路径
images_dir = r'D:\pythonProjects\yuanshenAI\datasets\yuanshen\images'
labels_dir = r'D:\pythonProjects\yuanshenAI\datasets\yuanshen\labels'

# 获取两个目录中的文件列表
images_files = os.listdir(images_dir)
labels_files = os.listdir(labels_dir)

# 创建两个集合，存储文件名
images_set = set(images_files)
labels_set = set(labels_files)

#去掉后缀
images_set = {os.path.splitext(file)[0] for file in images_files}
labels_set = {os.path.splitext(file)[0] for file in labels_files}

# 找出images目录中有但labels目录中没有的文件
only_in_images = images_set - labels_set

# 找出labels目录中有但images目录中没有的文件
only_in_labels = labels_set - images_set

# 输出结果
print(f"Files only in images directory:\n{only_in_images}")
print(f"Files only in labels directory:\n{only_in_labels}")