标签：xml box VOC int list __ childNodes data

"""
function of the code: resize_box

author@bjtu_huangyuxiang
"""
from __future__ import division
import os
import xml.dom.minidom as xdm

def resize_box(AnnoPath,OutPath):

    files = os.listdir(AnnoPath)
    i = 0
    for xmlfile in files:
        DomTree = xdm.parse(AnnoPath+xmlfile)
        root = DomTree.documentElement #获得xml文档对象
        objectlist = root.getElementsByTagName('object') #获得xml文档目标名称

        for objects in objectlist:
            bndbox = objects.getElementsByTagName('bndbox')
            for box in bndbox:
                x1_list = box.getElementsByTagName('xmin')
                y1_list = box.getElementsByTagName('ymin')
                x2_list = box.getElementsByTagName('xmax')
                y2_list = box.getElementsByTagName('ymax')
                xmin = int(x1_list[0].childNodes[0].data)
                ymin = int(y1_list[0].childNodes[0].data)
                xmax = int(x2_list[0].childNodes[0].data)
                ymax = int(y2_list[0].childNodes[0].data)
            "下列四行代码是调整后坐标值与未调整前的坐标关系"
            x1min = xmin+0.25*(xmax-xmin)
            y1min = ymin+0.25*(ymax-ymin)
            x1max = xmin+0.7*(xmax-xmin)
            y1max = ymin+0.75*(ymax-ymin)

            x1_list[0].childNodes[0].data = int(x1min)
            y1_list[0].childNodes[0].data = int(y1min)
            x2_list[0].childNodes[0].data = int(x1max)
            y2_list[0].childNodes[0].data = int(y1max)
            if not os.path.exists(OutPath):
                os.makedirs(OutPath)
            save_file = OutPath + xmlfile
            with open(save_file,'w') as newxml:
                DomTree.writexml(newxml)
                print("{}重写入完成".format(xmlfile))
        i += 1
    print("共计修改%s个xml文件" % i)

"-----------------------main function-------------------------"
if __name__ == "__main__":
    AnnoPath = r'D:\desktop\hat_test\xml/'
    OutPath = r'D:\desktop\hat_test\newxml/'
    resize_box(AnnoPath, OutPath)

标签：xml,box,VOC,int,list,__,childNodes,data
来源： https://blog.csdn.net/weixin_46716951/article/details/122380056

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。