删除xml或者图像文件夹多余的文件

#encoding:utf-8

import os  #os：操作系统相关的信息模块

#存放原始图片地址
data_img_dir = "./faceimage/"
#存放原始xml地址
data_xml_dir="./faceimage_xml/"
#建立列表，用于保存图片信息
jpg_list=[]
#建立列表，用于保存xml信息
xml_list=[]

for file in os.listdir(data_img_dir): #file为current_dir当前目录下图片名
    if file.endswith(".jpg"): #如果file以jpg结尾
        file_name = file[:-4]    #只保留数字,去掉后缀
        jpg_list.append(file_name) #将write_name添加到file_list列表最后
#sorted(file_list) #将列表中所有元素随机排列
# number_of_lines = len(file_list) #列表中元素个数

#file 为文件下的xml文件
for file in os.listdir(data_xml_dir):
    if file.endswith(".xml"):
        file_name=file[:-4]
        xml_list.append(file_name)


jpg_list.sort()
xml_list.sort()
#将图片信息写入txt文件中，逐行写入
# for current_line in range(number_of_lines): 
#     write_file.write(file_list[current_line]+'\n')
# #关闭文件
# write_file.close()


# with open("./jpg.txt","r") as f:
#     for i in f:
#         # print(i)
#         jpg_list.append(i.strip('\n')) 


# with open("./xml.txt","r") as f:
#     for i in f:
#         # print(i)
#         xml_list.append(i.strip('\n')) 

# print(xml_list)
# print("------------------------------------------")
# print(jpg_list)

# 需要注意的是 list中文件较多的放前面,这样才能比较出两个list不同
print(set(xml_list).difference(set(jpg_list)))