深度学习训练前快速批量修改数据集中的图片名（按原顺序）

乱序排名

深度学习训练前快速批量修改数据集中的图片名，使用下面代码时只需要修改两个路径即可，见下：

具体代码见下：

import os
import shutil

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

    '''
    def __init__(self):
        self.path = 'Images/val/val2017'  #表示需要命名处理的文件夹
        self.save_path='Images/val/val2017 2'#保存重命名后的图片地址
    def rename(self):
        filelist = os.listdir(self.path) #获取文件路径
        total_num = len(filelist) #获取文件长度（个数）
        i = 1  #表示文件的命名是从1开始的
        for item in filelist:
            print(item)
            if item.endswith('.jpg'):  #初始的图片的格式为jpg格式的（或者源文件是png格式及其他格式，后面的转换格式就可以调整为自己需要的格式即可）
                src = os.path.join(os.path.abspath(self.path), item)#当前文件中图片的地址
                dst = os.path.join(os.path.abspath(self.save_path), ''+str(i) + '.jpg')#处理后文件的地址和名称,可以自己按照自己的要求改进
                try:
                    #os.rename(src, dst)
                    shutil.copy(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()

按原图片顺序修改图片名

上面提供方法修改后图片名是乱序的，没有按照原图片顺序修改，如果需要按原图片排序修改，详见我的另外一篇博文，链接为：添加链接描述

主要就在上面代码基础上加一行代码即可，代码见下：

# filelist.sort(key=lambda x: int(x.split('.')[0])) # 先按照名称顺序排好序
filelist.sort(key=lambda x: int(x.split('(')[1].split(')')[0]))  # 先按照名称顺序排好序

总结

以上就是快速批量修改数据集中图片名的方法，希望能帮助到你，多多支持，谢谢！