predict_generator
毕竟并没有拖延我的预测。问题出在__getitem__
方法上。例如,使用n_batch=32
,该方法将产生从1到32,然后从2到33等的值,而不是从1到32、33到64等。
def __getitem__(self, idx):
# batch_x is a numpy.ndarray
idx_min = idx*self.batch_size
idx_max = min(idx_min + self.batch_size, self.n)
batch_x = (
self.images[idx_min:idx_max]
.concatenate()
.reshape(self.batch_size, 720, 1280, 1)
)
batch_y = self.hf[idx_min:idx_max]