dxy1018的博客

原创 输入维度不匹配问题

程序运行报错问题：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]即可。还有当时我是积分