结合之前的博客: http://www.cnblogs.com/Allen-rg/p/5834551.html#3949333
用caffemodel去测试单通道的图像(mnist数据集)时,出现了问题,例如:维度不匹配、单通道图像和三通道图像不能强制转换等问题。
因为链接中的代码适用于对RGB三通道的图像的分类。
只需要将代码中: im
=
caffe.io.load_image(img)
#加载图片
修改为: im=caffe.io.load_image(img,False) 即可
将图像读入修改为单通道的图像读入即可;
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:在caffe中用训练好的 caffemodel 来分类新的图片所遇到的问题 - Python技术站