推荐一个好用的数据集处理脚本

前段时间在整理数据集，但是需要先处理一下数据集，批量的修改图片的名字，因此找了一下的代码，这非常好用，分享给大家。

注意事项：转换后，这脚本会把你之前文件夹的内容全删掉。

import osclass BatchRename():    '''    批量重命名文件夹中的图片文件    '''​    def __init__(self):        self.path = 'D:/Downloads/cod/鱼数据集/released-dataset/标注/image1'  # 表示需要命名处理的文件夹        self.save_path = 'D:/Downloads/cod/鱼数据集/released-dataset/标注/image'  # 保存重命名后的图片地址    def rename(self):        filelist = os.listdir(self.path)  # 获取文件路径        total_num = len(filelist)  # 获取文件长度（个数）        i = 0  # 表示文件的命名是从200000开始的        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)                    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()