生活本来就刺激 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
    关注

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

    评论

报告相同问题?

悬赏问题

  • ¥15 安卓adb backup备份应用数据失败
  • ¥15 eclipse运行项目时遇到的问题
  • ¥15 关于#c##的问题:最近需要用CAT工具Trados进行一些开发
  • ¥15 南大pa1 小游戏没有界面,并且报了如下错误,尝试过换显卡驱动,但是好像不行
  • ¥15 没有证书,nginx怎么反向代理到只能接受https的公网网站
  • ¥50 成都蓉城足球俱乐部小程序抢票
  • ¥15 yolov7训练自己的数据集
  • ¥15 esp8266与51单片机连接问题(标签-单片机|关键词-串口)(相关搜索:51单片机|单片机|测试代码)
  • ¥15 电力市场出清matlab yalmip kkt 双层优化问题
  • ¥30 ros小车路径规划实现不了,如何解决?(操作系统-ubuntu)