Python批量修改xml的坐标值全部转为整数的实例代码
(编辑:jimmy 日期: 2024/11/16 浏览:3 次 )
发现一个有意思的现象,labelimg打开图片和xml标签时候,看不到标注好的框框,仔细查看了xml文件,没发现什么异常,后面试一下,才发现是不能识别xml里的坐标值有小数点的情况。只能四舍五入都转成整数。
如:
<bndbox> <xmin>1404.35</xmin> <ymin>0</ymin> <xmax>1458.56</xmax> <ymax>111.96</ymax> </bndbox>
# *_* coding : UTF-8 *_* # 开发人员: csu·pan-_-|| # 开发时间: 2020/11/25 21:42 # 文件名称: resize_img_xml.py # 开发工具: PyCharm # 功能描述: 改变xml文件里的某个属性 import os import xml.etree.ElementTree as ET path = r'C:\Users\Administrator\Desktop\JPEGImages' # 包含xml的文件夹路径 def edit_xml(xml_file): """ 修改xml文件 :param xml_file:xml文件的路径 :return: """ all_xml_file = os.path.join(path, xml_file) tree = ET.parse(all_xml_file) print(all_xml_file) objs = tree.findall('object') for ix, obj in enumerate(objs): obj_bnd = obj.find('bndbox') obj_xmin = obj_bnd.find('xmin') obj_ymin = obj_bnd.find('ymin') obj_xmax = obj_bnd.find('xmax') obj_ymax = obj_bnd.find('ymax') xmin = float(obj_xmin.text) ymin = float(obj_ymin.text) xmax = float(obj_xmax.text) ymax = float(obj_ymax.text) obj_xmin.text = str(round(xmin)) #四舍五入转为整数 obj_ymin.text = str(round(ymin)) obj_xmax.text = str(round(xmax)) obj_ymax.text = str(round(ymax)) tree.write(all_xml_file, method='xml', encoding='utf-8') # 覆盖更新xml文件 if __name__ == '__main__': files = os.listdir(path) # 获取文件名列表 for i, file in enumerate(files): if file.endswith('.xml'): edit_xml(file)
转完之后,标签就可以正常看到了
下一篇:基于Python采集爬取微信公众号历史数据