Python图像/数据集批量打乱+重命名（附python代码）

所需python环境

torch
os
random

直接上代码

（循环里的“000”表示修改后的前缀加000，可灵活设置）

import os
import random


class ImageRename():
    def __init__(self):
        self.path = '/data01/home/scy0658/run/Berber/datasets'  # 图片所在文件夹

    def rename(self):
        filelist = os.listdir(self.path)
        random.shuffle(filelist)
        total_num = len(filelist)

        i = 0

        for item in filelist:
            if item.endswith('.png'):
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), '000' + format(str(i), '0>3s') + '.png')
                os.rename(src, dst)
                print('converting %s to %s ...' % (src, dst))
                i = i + 1
        print('total %d to rename & converted %d pngs' % (total_num, i))


if __name__ == '__main__':
    newname = ImageRename()
    newname.rename()

效果图：

（假设这是乱的） 

（我是整齐的）

 

 注：代码部分原网址丢失，小伙伴看到可以踢我一下。