当使用keras.utils.Sequence时，keras预言生成器正在改组其输出

当使用keras.utils.Sequence时，keras预言生成器正在改组其输出

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]