使用python批量修改XML文件中图像的depth值资源-CSDN文库

144 浏览量 2020-12-17 05:44:15 上传评论收藏 72KB PDF 举报

资源详情

资源评论

资源推荐

使用使用python批量修改批量修改XML文件中图像的文件中图像的depth值值

最近刚刚接触深度学习，并尝试学习制作数据集，制作过程中发现了一个问题，现在跟大家分享一下。问题是这样的，在制作

voc数据集时，我采集的是灰度图像，并已经用labelimg生成了每张图像对应的XML文件。训练时发现好多目标检测模型使用

的训练集是彩色图像，因此特征提取网络的输入是m×m×3的维度的图像。所以我就想着把我采集的灰度图像的深度也改成3

吧。批量修改了图像的深度后，发现XML中的depth也要由1改成3才行。如果重新对图像标注一遍生成XML文件的话太麻烦，

所以就想用python批量处理一下。果然在网上找到了类似的代码，简单修改一下就可以实现我们想要的功能了。

全部代码如下

#coding:utf-8

import os

import os.path

import xml.dom.minidom

path='E:/data/ann/'#这里修改为自己存放XML文件的路径

files=os.listdir(path) #获取路径下的所有文件的名称

s=[] for xmlFile in files:

if not os.path.isdir(xmlFile): #判断是否是文件夹，不是文件夹才打开

print(xmlFile)

#将获取到的xml文件名送入到dom解析

dom=xml.dom.minidom.parse(os.path.join(path,xmlFile))

root=dom.documentElement

###获取标签对depth之间的值

depth=root.getElementsByTagName('depth')

#修改相应标签的值

for i in range(len(depth)):

print(depth[i].firstChild.data)

a = depth[i].firstChild.data

print(type(a))

depth[i].firstChild.data=3

print(depth[i].firstChild.data)

#保存修改到xml文件中

with open(os.path.join(path,xmlFile),'w') as fh:

dom.writexml(fh)

print('修改depth成功！')

上面的代码的思路是，读取XML文件，并修改depth节点的内容修改为3，通过循环读取XML文件，实现批量化修改XML文件

中depth的值。

修改前后的结果

XML修改前depth的值：

本内容试读结束，登录后可阅读更多

下载后可阅读完整内容，剩余1页未读，立即下载

评论0

内容反馈

weixin_38628612

粉丝: 8
资源: 942

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈

feedback-tip