在使用pycaffe定义网络、使用pycaffe进行网络训练与测试之后得到caffemodel
文件,下面利用caffemodel
进行分类:
- 导入相关库
import caffe
- 配置
# caffemodel文件
MODEL_FILE = 'model/_iter_10000.caffemodel'
# deploy文件,参考/caffe/models/bvlc_alexnet/deploy.prototxt
DEPLOY_FILE = 'deploy.prototxt'
# 测试图片存放文件夹
TEST_ROOT = 'datas/'