python提取xml的所有框坐标_Python批量修改xml的坐标值全部转为整数的实例代码

当使用labelimg工具打开包含小数点坐标的XML文件时,无法正确显示标注框。解决方法是将XML文件中所有坐标值四舍五入转换为整数。本文提供了一个Python脚本,用于读取XML文件,找到object中的bndbox,将xmin、ymin、xmax和ymax的值转换为整数,并覆盖保存XML文件。
摘要由CSDN通过智能技术生成

发现一个有意思的现象,labelimg打开图片和xml标签时候,看不到标注好的框框,仔细查看了xml文件,没发现什么异常,后面试一下,才发现是不能识别xml里的坐标值有小数点的情况。只能四舍五入都转成整数。qFt免费资源网

如:

qFt免费资源网

1404.35

0

1458.56

111.96

# *_* 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)

转完之后,标签就可以正常看到了

qFt免费资源网

qFt免费资源网

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值