刪除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)))