一、问题提出
上一篇博客,博主用了一个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()
三、效果