003_xml中出现浮点数变为整数

最新推荐文章于 2024-07-22 09:40:17 发布

原创最新推荐文章于 2024-07-22 09:40:17 发布 · 430 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

数据集处理专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种Python脚本，用于批量修正XML标注文件中边界框坐标的精度问题，通过四舍五入的方式将浮点坐标转换为整数坐标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# *_* coding : UTF-8 *_*



import os
import xml.etree.ElementTree as ET

path = r'D:\03_date_work\ann'  # 包含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)