生活本来就刺激
2018-12-17 05:22
采纳率: 0%
浏览 4.5k

keras.util.sequence + fit_generator 如何实现多输出model

输入输出的形式是下面这样:

model = Model(inputs=input_img, outputs=[mask,net2_opt,net3_opt])

由于sequence要求一定要返回一个两个参数的远足,所以生成器的_getitem_的实现如下:

class DataGenerator(keras.utils.Sequence):
    def __getitem__(self, index):
        #生成每个batch数据,这里就根据自己对数据的读取方式进行发挥了
        # 生成batch_size个索引
        batch_indexs = self.indexes[index*self.batch_size:(index+1)*self.batch_size]
        # 根据索引获取datas集合中的数据
        batch_datas = [self.datas[k] for k in batch_indexs]

        # 生成数据
        images, masks,heatmaps,xyzs = self.data_generation(batch_datas)

        return (images, [masks,heatmaps,xyzs])

output中的mask并不能与getitem的返回值匹配。

会报错:
ValueError: Error when checking target: expected conv_1x1_x14 to have 4 dimensions, but got array with shape (3,1)

请问,是不是keras.util.sequence不能实现多输出问题?

  • 写回答
  • 好问题 提建议
  • 追加酬金
  • 关注问题
  • 邀请回答

1条回答 默认 最新

相关推荐 更多相似问题