解决目标检测数据集jpg和xml文件数量不匹配问题

前言

在yolo目标检测标记数据时，需要转coco格式，不是每一张图片都含有标注，需要把不含标注的jpg文件删除，运行下面代码即可

代码

import os
jpg_path = 'C:/Users/Desktop/jpg' #放.jpg的文件夹
jpg_path = 'C:/Users/Desktop/xml' #放.xml的文件夹
jpg_lis = os.listdir(jpg_path)
xml_lis = os.listdir(xml_path)
xml_name = [i.split('.')[0] for i in xml_lis]
for i in jpg_lis:
    if i.split('.')[0] not in xml_name:
        os.remove(jpg_path + i)