我正在尝试在训练有素的网上做出预测,但是我收到了这个错误:不能从图层’fc6’复制param 0权重
这是我的deploy.prototxt
名称:“CaffeNet”层{ 名称:“数据” 类型:“输入”…
你确定你正在训练150 x 227像素的图像吗?好像你只提供了一个 crop_size 参数和你的输入图像实际上是150乘150。 运用 "InnerProduct" 图层不允许您更改输入大小。使用您收到的错误消息进行此类更改。 检查您的培训日志并查看不同层的输出大小,以验证确实如此。
crop_size
"InnerProduct"
当形状不匹配时,通常会发生此错误。根据您的train_val.prototxt文件,您的训练数据集大小可能是(50)x1x150x150而不是150x227。
在deploy.prototxt中,4个输入维度是N,C,H,W(1,1,150,227),其中N是测试时要使用的图像样本数(可以根据您的方便进行修改),C是通道数(1表示灰度,3表示RGB),H是图像的高度,W是图像的宽度。