您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

当使用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]
其他 2022/1/1 18:44:58 有536人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶