如何用训练好的caffemodel测试图片分类呢?caffe官方提供python的分类例子在examples/00-classification.ipynb下,但是只介绍了caffe.Net分类,这篇博客记录了另一中方式caffe.Classifier,并介绍了两者的区别。
1. caffe.Net
步骤大概分为:模型导入->数据处理->分类和输出
模型导入
需要deploy(网络定义)和caffemodel(权值)文件。
model_def = caffe_root + 'models/bvlc_reference_caffenet/deploy.prototxt'
model_weights = caffe_root + 'models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel'
net = caffe.Net(model_def, # defines the structure of the model
model_weights, # contains the trained weights
caffe.TEST) # use test mode (e.g., don't perform dropout)
数据处理
首先确定均值,可以