caffe记录错误

  • 报错:source param shape is 20 3 5 5 ;target param shape is 20 1 5 5
    用caffe中mnist网络训练自己的数据得到的caffemodel,每次运行会得到不一样的结果。之后预测时将自己生成的deploy.prototxt也同样替换成mnist中的lent.prototxt,就发生这个报错。
    原因:原始shape不一致,同时又很明确的指出了是cov1层出现的错误。所以直接找shape\cpnv1,之后才发现是训练模型文件与模型定义文件中的shape不相符,我训练时图像库中有的图像是一个通道,有的是使用的3个通道,所以默认使用三个通道,而我的模型定义文件中的shape,其通道数写的是1个通道,故出现错误。
    minst自带的图片集是黑白两色的,每个像素值只有一个通道,而我的图片虽然也是黑白色,但是其实是RGB格式的。
    解决方案将lent.prototxt中Input层中的第二个dim改成3.

你可能感兴趣的:(caffe学习笔记)