python - 对于相同的模型架构,Keras 模型和 Caffe 模型的输出形状不同
问题描述
在 Keras 中,图像的输出形状在应用Convolution2D
Layer 后会改变 1,但当我们将相同的模型与 Caffe 应用时,它就可以工作。我正在复制IS&T 2017论文的结果。在那篇论文中,他们使用具有类似模型的 Caffe。
我的模型摘要:
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
conv2d_1 (Conv2D) (None, 61, 61, 32) 1568
_________________________________________________________________
**max_pooling2d_1 (MaxPooling2 (None, 30, 30, 32) 0**
_________________________________________________________________
conv2d_2 (Conv2D) (None, 26, 26, 48) 38448
_________________________________________________________________
max_pooling2d_2 (MaxPooling2 (None, 13, 13, 48) 0
_________________________________________________________________
conv2d_3 (Conv2D) (None, 9, 9, 64) 76864
_________________________________________________________________
max_pooling2d_3 (MaxPooling2 (None, 4, 4, 64) 0
---------------------------------------------------------------------
这导致此错误
ValueError: Negative dimension size caused by subtracting 5 from 4 for 'conv2d_4/convolution' (op: 'Conv2D') with input shapes: [?,4,4,64], [5,5,64,128].
所有尺寸和补丁大小均相同。
解决方案
推荐阅读
- python - 如何将下面显示的 excel 函数转换为 python pandas 代码?
- delphi - 是否有可能找出连接是否被计量?
- python - 使用 Matplotlib 和 Python 绘制两个直方图
- python - 如何从 html 页面中排除内容并仅保留 html 标签?
- ios - UINavigationItem 在 UINavigationBar 后面
- apache-spark - 如何为具有多个分区的 Kafka 主题启动多个消费者?
- php - 无法在 NGNIX 的服务器端生成 pdf
- laravel - 调度操作后获取 vuex 存储状态
- firebase - firebase admin nodejs 上的错误需要权限 iam.serviceAccounts.signBlob
- go - 如何在特定数据量中写入文件