图像数据集之python批量重命名

在训练模型之前要对数据集进行规范化，此文提供了简便易行的python代码，对图像数据集进行批量重命名。同时，举一反三，可对其他文件类型进行批量重命名

步骤：
1）将需要重命名的图片文件放入“picture”文件夹；
2）相同路径下，创建py文件，复制以下代码；
3）编译执行，完成批量重命名。

import os

def rename():
    # 原始图片路径
    path = r'picture'
    # 获取该路径下所有图片
    filelist = os.listdir(path)
    a = 1
    for files in filelist:
        # 原始路径
        Olddir = os.path.join(path, files)

        # if os.path.isdir(Olddir):
        #	continue
        # 将图片名切片,比如 xxx.bmp 切成xxx和.bmp
        # xxx
        filename = os.path.splitext(files)[0]
        # .bmp
        filetype = os.path.splitext(files)[1]
        # 需要存储的路径 a 是需要定义修改的文件名
        Newdir = os.path.join(path, str(a) + filetype)
        os.rename(Olddir, Newdir)
        a += 1


rename()

效果