程序运行报错问题:Dimension 0 in both shapes must be equal, but are 1 and 512. Shapes are [1,1,128,512] and [512,256,1,1].
原因:当我采用预训练模型resnet50_weights_tf_dim_ordering_tf_kernels_notop.h5文件载入时出现上述错误,意思是两个矩阵的第一维不一样,一个是1一个是512.修改为[1,1,128,512]和[1,256,1,1]即可。
还有当时我是积分下载的这个模型,故可能有错,建议还是从官方网站上下载,就是比较慢.
官方网站:https://github.com/fchollet/deep-learning-models/releases
tf是tensorflow做后端
th是theano做后端
输入维度不匹配问题
最新推荐文章于 2022-11-08 20:12:39 发布