windows10+cuda7.5+caffe

参考:http://blog.csdn.net/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 就可以了






  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值