Keras实例:generator生成器(多输入/ImageDataGenerator)

当我们在训练模型的时候,有时会因为数据太大造成OOM或者影响效率,generator是Keras的一种解决方案,不仅如此,generator还能在训练时根据需要实时进行数据增强,这次我们继续使用Keras多输入模型实例的代码。

生成器的关键是yield,它能在循环中不停的返回数据,每次返回之前,我们都可以对数据进行操作,比如增加噪音。善用此功能可以增加模型的鲁棒性。首先我们用一个while循环无限跑,因为我们并不知道一共要训练多少次数据。然后每次循环都返回下一组batch_size个数据。与之前不同的是,我们这次使用了dict而不是数组来产生input,dict的key必须与模型中Input的name参数一致。

def data_generator(x1, x2, y, batch_size):
	size = len(y)
	while 1:
		for i in range(int(size / batch_size)):
			in1 = x1[i*batch_size: (i+1)*batch_size]
			in2 = x2[i*batch_size: (i+1)*batch_size]
			out = y[i*batch_size: (i
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值