在使用keras(Tensorflow后端)跑yolo v2时,报的错误,
1 该错误出现位置在训练函数fit_generator()中
model.fit_generator(generator = train_batch,
steps_per_epoch = len(train_batch),
epochs = 100,
verbose = 1,
validation_data = valid_batch,
validation_steps = len(val_batch),
callbacks = [early_stop, checkpoint, tensorboard],
max_queue_size = 3)
2 解决方案有如下两种:
- 将verbose = 1 改为verbose=0,即可
- 如果steps_per_epoch = 0, 就会报这个错误,所以确保batch_size < dataset size,解决方案,想办法不让steps_per_epoch不为0即可。