u014515978
生活本来就刺激
2018-12-17 05:22

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条回答

为你推荐

换一换