结合之前的博客: 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) 即可
将图像读入修改为单通道的图像读入即可;