【深度学习数据集】打乱图片并批量改名

一、问题提出

上一篇博客，博主用了一个python脚本按帧提取了视频文件制作成图片数据集，此时，两种图片之间的相似度是比较高的。在深度学习中，我们要求图片最好是随机的，不要有规律。而且为了操作方便，我们也要将数据集按一定顺序改名。

 

二、代码实现

# -*- coding:utf-8 -*-

import os
import random
class ImageRename():
    def __init__(self):
        self.path = 'F:\\dataset\\G0\\image'#图片所在文件夹

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

        i = 0

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

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

三、效果