在caffe中用训练好的 caffemodel 来分类新的图片所遇到的问题

结合之前的博客: 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) 即可

将图像读入修改为单通道的图像读入即可;

参考博客

原文地址:https://www.cnblogs.com/Allen-rg/p/8805534.html