Keras中使用ImageDataGenerator对每一个batch进行数据增强,再使用train_on_batch

【时间】2019.12.17

【题目】Keras中使用ImageDataGenerator对每一个batch进行数据增强,再使用train_on_batch

一、使用ImageDataGenerator的flow方法可以获得一个生成器,之后可以使用__next__()或者next()获取每个batch的数据(x,y),再传入train_on_batch进行训练。

import time
from keras.preprocessing.image import ImageDataGenerator
import numpy as np

imgs=np.random.randint(0,10,size=(7,100,100,3))

datagen = ImageDataGenerator(
    featurewise_center=True,
    featurewise_std_normalization=True,
    rotation_range=20,
    width_shift_range=0.2,
    height_shift_range=0.2,
    horizontal_flip=True)
label=np.random.randint(0,10,size=(7,10))
f=datagen.flow(imgs,[0,1,2,3,4,5,6],batch_size=3)

# print(f.next()[1])
# time.sleep(2)
# print(f.next()[1])
# time.sleep(2)
# print(f.next()[1])
i=0
while True:
    if i==10:
        break
    time.sleep(1)
    i +=1
    x,y=next(f)
    print(x.shape,y)

参考:

keras:ImageDataGenerator的flow方法

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值