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]
