A `Concatenate` layer requires inputs with matching shapes except for the concat axis.

在跑别人的以keras为深度学习框架的代码时,遇到错误:A Concatenate layer requires inputs with matching shapes except for the concat axis。这主要就是图像的通道数的位置引发的问题。
首先我们要知道,安装好的keras框架,它的后端有两种:tensorflow和theano。
在ubuntu环境下,命令行输入vim ~/.keras/keras.json可以查看。
A `Concatenate` layer requires inputs with matching shapes except for the concat axis._第1张图片
我的电脑以‘tensorflow’作为backend。你也可以修改切换至theano后端。
A `Concatenate` layer requires inputs with matching shapes except for the concat axis._第2张图片
在tensorflow中图像格式为:高,宽,通道数,也就是“channels_last”模式。
在theano中图像格式为:通道数,高,宽,也就是“channels_first”模式。
在不同的后端下,图像通道数的位置不同,代码要有相应改动。
我的出错代码是这样的:

报错是这样的:

Got inputs shapes: [(None, 32, 32, 512), (None, 32, 32, 256)]。我的图像通道数在axis=3的位置,而别人的代码中却把axis设为1,说明他用theono作为后端。此处,你只需把代码中axis改为3即可运行成功。(注意:我用tensorflow作为后端)

然后我的代码就顺利运行了。

你可能感兴趣的:(A `Concatenate` layer requires inputs with matching shapes except for the concat axis.)