首页 > 其他分享 >keras:ImageDataGenerator的flow_from_directory方法

keras:ImageDataGenerator的flow_from_directory方法

时间:2022-10-27 13:38:08浏览次数:56  
标签:shuffle xm flow ImageDataGenerator batch 索引 directory size


结合上文,​​flow的ImageDataGenerator​​的分析,该部分伪代码如下:

xm,y=getDataIndex()#获取所有文件夹中所有图片索引,以及文件夹名也即标签

while(True):
if shuffle==True:
shuffle(xm,y)#打乱图片索引及其标签
for i in range(0,len(x),batch_size):
xm_batch=xm[i:i+batch_size]#文件索引
y_batch=y[i:i+batch_size]
x_batch=getImg(xm_batch)#根据文件索引,获取图像数据
ImagePro(x_batch)#数据增强
#保存提升后的图片
#saveToFile()
yield

源码参考:​​/Lib/site-packages/keras/preprocessing/image.py​


标签:shuffle,xm,flow,ImageDataGenerator,batch,索引,directory,size
From: https://blog.51cto.com/u_15847885/5800880

相关文章