【Python】批量修改图片文件名和xml文件信息

Python 同时被 2 个专栏收录
6 篇文章 0 订阅
4 篇文章 0 订阅

在使用tensorflow进行数据训练时,由于原图片文件名较繁琐,且由于根据原图片名生成的xml标签文件中生成了包含filename的标签属性,不利于后期测试训练效果,故通过Python代码对图片名和xml文件信息进行批量修改为由0开始的顺序递增文件名,以配合后期的测试工作,在此记录下。首先,这是图片名称和xml文件信息:

图片文件名:

标签文件名:

xml信息:

Python代码如下:

#coding:utf-8
from PIL import Image
import os.path
import glob
import xml.etree.ElementTree as ET
import xml.dom.minidom

i = 0
xmldir = "E:\\object-detection-Game-2018-5-31\\xmls"
imgsdir = "E:\\object-detection-Game-2018-5-31\\imgs"
outdir = "E:\\object-detection-Game-2018-5-31\\imgs"
for xmlfile in os.listdir(xmldir):
    xmlname = os.path.splitext(xmlfile)[0]
    for pngfile in os.listdir(imgsdir):
        pngname = os.path.splitext(pngfile)[0]
        if pngname == xmlname:
             # 修改图片文件名
             # 图片文件名修改前后的路径
             olddir = os.path.join(os.path.abspath(imgsdir), pngname + ".png")
             newdir = os.path.join(os.path.abspath(imgsdir), str(i)+".png")
             os.rename(olddir, newdir)
             print(xmlfile, '----->', str(i) + '.png')
             # 修改filename结点属性
             # 读取xml文件
             dom = xml.dom.minidom.parse(os.path.join(xmldir, xmlfile))
             root = dom.documentElement

             # 获取标签对filename之间的值并赋予新值i
             root.getElementsByTagName('filename')[0].firstChild.data = str(i) + '.png'

             # 将修改后的xml文件保存
             # xml文件修改前后的路径
             old_xmldir = os.path.join(xmldir, xmlfile)
             new_xmldir = os.path.join(xmldir, str(i)+'.xml')
             # 打开并写入
             with open(old_xmldir, 'w') as fh:
                 dom.writexml(fh)
             os.rename(old_xmldir, new_xmldir)
             i += 1
print('total number is ', i)

运行结果如下:

修改后的文件信息:

 

reference:

https://blog.csdn.net/LOVE1055259415/article/details/79166754

https://blog.csdn.net/wklken/article/details/7603071

 

 

 

  • 5
    点赞
  • 9
    评论
  • 13
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值