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

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条回答 默认 最新

  • 梅川鸡尾酒 2020-09-16 14:11
    关注

    可以实现,你的问题不在于多输入多输出上,在于输出维度

    评论

报告相同问题?

悬赏问题

  • ¥50 Kubernetes&Fission&Eleasticsearch
  • ¥15 有没有帮写代码做实验仿真的
  • ¥15 報錯:Person is not mapped,如何解決?
  • ¥30 vmware exsi重置后登不上
  • ¥15 易盾点选的cb参数怎么解啊
  • ¥15 MATLAB运行显示错误,如何解决?
  • ¥15 c++头文件不能识别CDialog
  • ¥15 Excel发现不可读取的内容
  • ¥15 关于#stm32#的问题:CANOpen的PDO同步传输问题
  • ¥20 yolov5自定义Prune报错,如何解决?