参考:http://blog.youkuaiyun.com/baidu_26408419/article/details/53711640
http://www.itdadao.com/articles/c15a262087p0.html
1、anaconda2
2、cuda7.5
3、cudnn5(就是一个压缩包)
4、windows-caffe
https://github.com/Microsoft/caffe
5、在vs中编译
修改compute_image_mean.exe 添加cuda路径 添加python路径
6、添加环境变量
D:\caffe\caffe-master\Build\x64\Release
7、训练模型
下载mnist数据集训练模型
8、测试
参考:http://nbviewer.jupyter.org/github/BVLC/caffe/blob/master/examples/00-classification.ipynb
用caffe的python接口调用训练好的模型
(1)计算均值文件./binaryproto
目录./caffe-master/Build/x64/Release 下的工具 compute_image_mean.exe
命令行: compute_image_mean.exe *.lmdb *.binaryproto
(2)binaryproto转npy
# -*- coding: utf-8 -*-
import caffe
import numpy as np
MEAN_PROTO_PATH = './mnist_mean.binaryproto' # 待转换的pb格式图像均值文件路径
MEAN_NPY_PATH = './mnist_mean.npy' # 转换后的numpy格式图像均值文件路径
blob = caffe.proto.caffe_pb2.BlobProto() # 创建protobuf blob
data = open(MEAN_PROTO_PATH, 'rb' ).read() # 读入mean.binaryproto文件内容
blob.ParseFromString(data) # 解析文件内容到blob
array = np.array(caffe.io.blobproto_to_array(blob)) # 将blob中的均值转换成numpy格式,array的shape (mean_number,channel, hight, width)
mean_npy = array[0] # 一个array中可以有多组均值存在,故需要通过下标选择其中一组均值
np.save(MEAN_NPY_PATH ,mean_npy)
(3)测试
在导入cv2模块的发现没有
早python3里面有cv2 但是python2里面没有 用了pip install opencv-python 也没有用
解决方法:激活annconda2的环境: 运行D:\ProgrammeFiles\Anaconda2\Scripts\activate.bat 然后再pip install 就可以了