自己做的网站如何被百度检索,wordpress 更新 ftp,百度移动应用,wordpress 项目需求#xff1a; 数据集的数据增强中#xff0c;有时需要将xml标注文件中的类别做修改为新类别#xff0c;或者将几个类别合并为一个类别。 解决方法#xff1a; 使用pythonimport xml.etree.ElementTree将xml标注文件中的类别名称做修改。代码如下…需求 数据集的数据增强中有时需要将xml标注文件中的类别做修改为新类别或者将几个类别合并为一个类别。 解决方法 使用pythonimport xml.etree.ElementTree将xml标注文件中的类别名称做修改。代码如下
import os
import glob
import xml.etree.ElementTree as etdef change_labels(source_dir):file_list glob.glob(os.path.join(source_dir,*.xml))print(len(file_list))index 1for file in file_list:print(file,index)index 1tree_ et.ElementTree()tree_.parse(file)root et.Element(annotation)folder et.SubElement(root, folder)folder.text imagesfilename et.SubElement(root, filename)filename.text tree_.find(.//filename).textpath et.SubElement(root, path)path.text /home/Desktop/labelssource et.SubElement(root, source)database et.SubElement(source, database)database.text Unknownsize et.SubElement(root, size)width et.SubElement(size, width)width.text tree_.find(.//width).textheight et.SubElement(size, height)height.text tree_.find(.//height).textdepth et.SubElement(size, depth)depth.text 3segmented et.SubElement(root, segmented)segmented.text 0objects tree_.findall(.//object)anchor 0for object in objects:try:name_ object.find(name).textexcept:print(file)print(-----------------------------------------)continuen if name_ van or name_ bus or name_ truck:n carelse:n name_object_ et.SubElement(root, object)name et.SubElement(object_, name)name.text npose et.SubElement(object_, pose)pose.text Unspecifiedtruncated et.SubElement(object_, truncated)truncated.text 0difficult et.SubElement(object_, difficult)difficult.text 0bndbox et.SubElement(object_, bndbox)xmin et.SubElement(bndbox, xmin)xmin.text object.find(.//xmin).textymin et.SubElement(bndbox, ymin)ymin.text object.find(.//ymin).textxmax et.SubElement(bndbox, xmax)xmax.text object.find(.//xmax).textymax et.SubElement(bndbox, ymax)ymax.text object.find(.//ymax).texttree et.ElementTree(root)tree.write(file,encodingutf-8)if __name__ __main__:source_dir /home/Desktop/labels/Annotationsflip_images(source_dir) 使用以上代码需要修改原图像和标注文件所在文件夹路径source_dir也需要修改类别上述代码将van”、“bus”、“truck”修改为“car”这里可根据实际情况自行修改。亲测可用。
