图片训练集预处理(随机抽取训练集+训练集编号)

第一步：随机抽取训练集

import os, random, shutil
def moveFile(fileDir):
        pathDir = os.listdir(fileDir)    #取图片的原始路径
        filenumber=len(pathDir)
        picknumber=900 #按照rate比例从文件夹中取一定数量图片
        sample = random.sample(pathDir, picknumber)  #随机选取picknumber数量的样本图片
        print (sample)
        for name in sample:
                shutil.move(fileDir+name, tarDir+name)
        return

if __name__ == '__main__':
	fileDir = "E:/source/"    #源图片文件夹路径
	tarDir = 'E:/result/'    #移动到新的文件夹路径
	moveFile(fileDir)

第二步：给训练集编号

import os 
import os.path
rootdir = "E:/result/"
files = os.listdir(rootdir)
b=1  
for name in files:
  a=os.path.splitext(name)
  #print a[0]
  newname = str(b)+'.jpg'
  b = b + 1
  os.rename(rootdir+name,rootdir+newname)