深度学习数据集常用处理程序

1-1. jpg&xml_Rename

#-----------------------------批量重命名图�?------------------------------------
#-----------重命名《图片》《xml文件》《任何格式的文件》都可以自定义序号来命名------------
import os
'''
该路径存储原始的图片，重命名后会覆盖掉原始的图片，所以在重命名之前选择复制一份，以免被失误重命名
建议在路径中使用：\\ 双斜
'''
path = 'D:\\Desktop\\wajueji-4672-4'
# 绝对路径
filelist = os.listdir(path)
'''
起始数字，重命名的第一个文件的名字会加1
'''
i = 0
# 仅用于数字开头的图片命名方法
for item in filelist:
        #print('item name is ',item)
        # jpg\png\bmp\xml  任何格式都支持，但是需要手动修改格式类
        if item.endswith('.jpg'):
                i = i + 1
                # 第一张图片命名为1.png
                name = str(i)
                # 将数字转换为字符串才能命�?
                src = os.path.join(os.path.abspath(path),item)
                # 原始图像的路�?
                dst = os.path.join(os.path.abspath(path), format(str(i), '0>4s') + '-E1-excavator.jpg')
                # 目标图像路径
        try:
                os.rename(src,dst)
                print('rename from %s to %s'%(src,dst))
                # 将转换结果在终端打印出来以便检�?
        except:
                continue

1-2.xml-lable-rename

"""
使用python xml解析树解析xml文件，批量修改xml文件里object节点下name节点的text
"""


import glob
import xml.etree.ElementTree as ET
path = r'D:\\Desktop\\wajueji-4672（后边数据有问题)\\clabel'    # xml文件夹路径
i = 0
for xml_file in glob.glob(path + '/*.xml'):
    # print(xml_file)
    tree = ET.parse(xml_file) 
    obj_list = tree.getroot().findall('object') 
    for per_obj in obj_list:
        if per_obj[0].text == 'wajueji':    # 错误的标签“33”
            per_obj[0].text = 'excavator'    # 修改成“44”
            i = i+1

    tree.write(xml_file)    # 将改好的文件重新写入，会覆盖原文件
print('共完成了{}处替换'.format(i))