Keras大坑

持续踩坑:

  1. 如果model.fit中的参数suffle=True时,会随机打算每一次epoch的数据。(默认打乱),但是验证数据默认不会打乱
  2. fit函数里,先执行validation_split 再 执行shuffle=True,所以val很有可能全是某一类样本了。
  3. 多图像输入的generator写法:
generator = ImageDataGenerator(
    featurewise_center=True,
    featurewise_std_normalization=True,
    rotation_range=20,
    width_shift_range=0.2,
    height_shift_range=0.2,
    horizontal_flip=True)

def generate_data_generator_for_two_images(X1, X2, Y):
    genX1 = generator.flow(X1,Y, seed=7)
    genX2 = generator.flow(X2, seed=7)
    while True:
            X1i = genX1.next()
            X2i = genX2.next()
            yield [X1i[0], X2i ], X1i[1]


model.fit_generator(generate_data_generator_for_two_images(audio, image, label),
                     steps_per_epoch=step, epochs=epochs,verbose=2, callbacks=callbacks_list,
                     validation_data=([audio_val,image_val],val_label),        
                     max_q_size=3*batch_size, workers=4, pickle_safe=False)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值