欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python批量更改图片名的方法

程序员文章站 2022-03-02 14:19:31
...

在图片数据集时,我们采集的图片往往是名字各异的,因此我们需要对整个文件夹里的图片名进行重命名。

如果手动进行修改过于麻烦,那么我们可以使用代码进行修改。

import os


class BatchRename():
    """
    批量重命名文件夹中的图片文件

    """

    def __init__(self):
        self.path = './train_img/stand_img'  # 表示需要命名处理的文件夹
        self.new_path = './train_img/stand_img'     # 表示图片处理后保存的文件夹,我选择同一文件夹

    def rename(self):
        filelist = os.listdir(self.path)  # 获取文件路径
        total_num = len(filelist)  # 获取文件长度(个数)
        i = 1  # 表示文件的命名是从1开始的
        for item in filelist:
            if item.endswith('.jpg'):  # 初始的图片的格式为jpg格式的(或者源文件是png格式及其他格式,后面的转换格式就可以调整为自己需要的格式即可)
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.new_path), '' + str(i) + '.jpg')  # 处理后的格式也为jpg格式的,当然这里可以改成png格式
                dst = os.path.join(os.path.abspath(self.path),
                                   'img' + format(str(i), '0>3s') + '.jpg')  # 这种情况下的命名格式为img000.jpg形式,可以自主定义想要的格式
                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                    i = i + 1
                except:
                    continue
        print('total %d to rename & converted %d jpgs' % (total_num, i))


if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()

 

相关标签: Python