网络编程
位置:首页>> 网络编程>> Python编程>> Python批量修改xml的坐标值全部转为整数的实例代码

Python批量修改xml的坐标值全部转为整数的实例代码

作者:一只有理想的喵(>^ω^<)  发布时间:2021-10-29 00:57:50 

标签:Python,批量修改,xml

发现一个有意思的现象,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批量修改xml的坐标值全部转为整数的实例代码

来源:https://blog.csdn.net/qq_36563273/article/details/110122238

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com