目标检测中删除不一致的xml和jpg文件

"""
删除image和xml不对应的文件
"""

import os
import shutil
# file_name_1 = r"10-11Image\img" #图片文件存放地址
# file_name_2 = r"10-11Image\xml" #标记文件存放地址
file_name_1 = r"C:\Users\admin\Desktop\hh-new\crop1280_1280" #图片文件存放地址
file_name_2 = r"C:\Users\admin\Desktop\hh-new\xml" #标记文件存放地址

image = []
annotation =[]
for file in os.listdir(file_name_1):
    if file.endswith(".jpg"):
        num, j = file.split(".")
        image.append(num)
for file in os.listdir(file_name_2):
    if file.endswith(".xml"):
        num, j = file.split(".")
        annotation.append(num)
for img in image:
    if img not in annotation:
        os.remove(os.path.join(file_name_1, str(img)+".jpg"))

for xml in annotation:
    if xml not in image:
        os.remove(os.path.join(file_name_2, str(xml)+".xml"))