本文主要是介绍Caffe学习:pycaffe利用caffemodel进行分类,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
在使用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/'
这篇关于Caffe学习:pycaffe利用caffemodel进行分类的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!