我的第一个二值化网络:BWN-XNOR-caffe安装注意项&&BWN-XNOR-caffe-master训练二进制cifar数据集
来源:互联网 发布:公考网络培训 编辑:程序博客网 时间:2024/05/16 14:24
转载自:http://blog.csdn.net/l297969586/article/details/62428426
终于找到一篇指导我运行的文章了,泪奔中
github地址https://github.com/loswensiana/BWN-XNOR-caffe
XNOR是用于权值与输入二值化的解决方案,以损失一部分精度为代价,来达到提高计算速度和减小缓存的目的
你需要提前准备好cuda配置与caffe环境配置.
具体可查看我之前博客:
ubuntu14.04+cuda7.5安装 官方步骤版
ubuntu14.04+cuda7.5+caffe+cudnn7.5+anaconda+opencv 2.4.9系统整合(2016.12.3)
从github上下载压缩文件后解压到home路径下,会出现一个BWN-XNOR-caffe-master文件夹.
由于我们的cudnn是v5版本的,而该工程是v3版本的,需要我们把之前caffe中的cudnn文件替换BWN-XNOR-caffe-master中的cudnn文件
在终端输入:
- 1
- 2
- 3
更改BWN-XNOR-caffe-master/src/caffe/ex_layers/binary_conv_layer.cu文件中的两个函数
110行cudnnConvolutionBackwardFilter_v3改为cudnnConvolutionBackwardFilter
128行cudnnConvolutionBackwardData_v3改为cudnnConvolutionBackwardData
注:这都是cudnn_v5版本函数名
之后建立BWN-XNOR-caffe-master/build文件夹
- 1
- 2
- 3
caffe编译完成
lilai619
2017-08-01 09:461楼- 你好,请教一下,你有测试xnor和没有二值化版本的速度吗?
l297969586
2017-08-01 09:56- 回复lilai619:我测试的一般都是python版本的。在tensorflow、借助lasagne的theano下做二值化卷积并没有达到论文里提到的接近64倍加速的效果
lilai619
2017-08-01 22:48- 回复l297969586:好的,很是感谢,有空我去尝试一下python版本的
续:
BWN-XNOR-caffe-master训练二进制cifar数据集
BWN-XNOR-caffe-master安装还有一些注意事项在上一个博客
目的:为了提高卷积过程的速度,和把caffe移植到FPGA上
- 1
- 1
- 2
- 3
修改examples/cifar10/train_nobias.sh
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
就是把– gpu $1 给删了
或者你在cifar10_full.nobias.solver文件中把最后两行的
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
训练
- 1
会得到两个文件
- 1
- 2
以下是在caffe-master中的过程
- 1
- 1
- 2
训练
- 1
会得到两个文件
- 1
- 2
- 3
检测
- 1
- 1
可能会出现的问题:
运行classify.py时报错 Mean shape incompatible with input shape
解决:
修改 ./python/caffe/io.py 文件
将
- 1
- 2
修改为:
- 1
- 2
- 3
- 4
- 5
- 6
- 7
即可.
以下为运行结果:
为了能够显示测试图片种类以及置信度,网上有位网友写了一个classify.py
地址:http://download.csdn.net/detail/caisenchuan/9513196
下载后替换Python/classify.py
因为这个classify.py增加了几个参数 –labels_file、–print_results、–force_grayscale可以显示预测结果,同时博主发现这个文件有问题,你直接运行是输出不了结果的,需要修改171行
- 1
- 2
修改为
- 1
- 2
之后cd到caffe-master,终端运行
- 1
显示结果
- 我的第一个二值化网络:BWN-XNOR-caffe安装注意项&&BWN-XNOR-caffe-master训练二进制cifar数据集
- BWN-XNOR-caffe-master训练二进制cifar数据集
- BWN-XNOR-caffe安装注意项
- ubuntu14.04安装theano的二进制网络theano-xnor-net
- Ubuntu+caffe训练cifar-10数据集
- caffe学习(二) CIFAR-10数据集上训练
- caffe训练CIFAR数据库
- caffe示例实现之1在CIFAR-10数据集上训练与测试Caffe
- 利用caffe训练cifar之训练数据集合生成leveldb
- Caffe学习记录:Cifar-10 自定义网络训练记录
- Caffe学习记录:Cifar-10 自定义网络训练记录
- cifar数据集基于Caffe的生成与测试
- 同或门(XNOR)电路的网络学习
- Windows Caffe 学习笔记(一)训练和测试CIFAR-10数据集
- caffe训练网络的实践
- caffe 训练测试自己的数据集
- caffe训练测试自己的数据集
- caffe训练自己的数据集
- 1074. 宇宙无敌加法器(20)
- struts、hibernate、spring、 mybatis、 spring boot 等面试题汇总
- Sublime Text 使用技巧
- PC访问profibus协议设备的方式
- Python Numpy 教程
- 我的第一个二值化网络:BWN-XNOR-caffe安装注意项&&BWN-XNOR-caffe-master训练二进制cifar数据集
- 513. Find Bottom Left Tree Value(BFS)
- 反射基础
- Servlet入门(13)
- Mac【安装SecureCRT客户端并激活】
- java基础知识之-----HashMap
- 关于c++程序运行的窗口停留问题
- 142. Linked List Cycle II
- java基础知识之-----HashTable