CNN的多个输入:图像和参数,如何合并

时间:2018-07-14 17:06:41

标签: python tensorflow keras concatenation conv-neural-network

我将Keras用于CNN,并且具有两种输入类型:对象的图像以及一个或两个以上描述对象的参数(例如重量)。如何使用两个数据源训练我的网络?串联似乎不起作用,因为输入具有不同的尺寸。我的想法是将图像分析的输出和参数以某种方式连接在一起,然后再将其发送到密集层中,但是我不确定如何做到。还是可以在Keras中合并两个分类,即对图像和参数进行分类,然后以某种方式合并分类?

1 个答案:

答案 0 :(得分:1)

您可以使用“并置”层合并两个输入。确保将多个输入转换为相同的形状;您可以通过在任一输入中添加额外的密集层来做到这一点,以便获得等长的末端层。在“连接”层中使用相同形状的输出。