目标检测数据集制作-批量移除空的xml文件

#  批量移除空的xml标注
import xml.etree.cElementTree as ET
import os

path_root = 'E:\\dataset\\PVEL-AD\\Detection\\test\\total_annotations\\'

xml_list = os.listdir(path_root)
for axml in xml_list:
    path_xml = os.path.join(path_root, axml)
    tree = ET.parse(path_xml)
    root = tree.getroot()

    if len(root.findall('object'))==0:
        tree.write(os.path.join('C:\\Users\\China\\Desktop\\y\\', axml))