提取xmls文件里指定标签到相应文件夹下

import os
import shutil
import xml.etree.ElementTree as ET

xmls_path = ''   # xmls
imgs_path = ''   # imgs
copy_path_imgs = ''
copy_path_xmls = ''
xml_fn_list = os.listdir(xmls_path)

count = 0
for i, xml_fn in enumerate(xml_fn_list):
    tree = ET.parse(os.path.join(xmls_path, xml_fn))
    root = tree.getroot()
    for element in root.findall('object'):
        name = element.find('name').text
        if name == 'aqmzc' or name == 'gzzc':
            print(name)
            count += 1
            ori_xml_path = os.path.join(xmls_path, xml_fn)
            img_fn = xml_fn.split('.')[0] + '.jpg'
            ori_img_path = os.path.join(imgs_path, img_fn)
            os.system("cp {} {}".format(ori_xml_path, copy_path_xmls))
            os.system("cp {} {}".format(ori_img_path, copy_path_imgs))
            
print('sucess', count)