博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
[caffe(一)]使用caffe训练mnist数据集
阅读量:5899 次
发布时间:2019-06-19

本文共 1476 字,大约阅读时间需要 4 分钟。

1.数据集的下载与转换

1)我们在mnist数据集上做测试,在这里下载这四个文件:

2)然后解压生成了以下四个文件:

3)数据格式转换

新建一个文本文档,更改后缀为 .bat  在里面复制以下code: 

..\..\Build\x64\Release\convert_mnist_data.exe  ..\..\data\mnist\train-images.idx3-ubyte ..\..\data\mnist\train-labels.idx1-ubyte ..\..\examples\mnist\mnist_train_lmdb  

echo.  

..\..\Build\x64\Release\convert_mnist_data.exe  ..\..\data\mnist\t10k-images.idx3-ubyte  ..\..\data\mnist\t10k-labels.idx1-ubyte ..\..\examples\mnist\mnist_test_lmdb 

pause  

其实第一个路径就是 exe文件所在路径, 后面两个就是数据所在路径,最后一个是输出文件路径

例如:

F:\caffe-master\caffe-master\Build\x64\Release\convert_mnist_data.exe E:\caffetest\mnist\mnist\train-images.idx3-ubyte E:\caffetest\mnist\mnist\train-labels.idx1-ubyte E:\caffetest\mnist\mnist\mnist_train_lmdb

echo.

F:\caffe-master\caffe-master\Build\x64\Release\convert_mnist_data.exe E:\caffetest\mnist\mnist\t10k-images.idx3-ubyte E:\caffetest\mnist\mnist\t10k-labels.idx1-ubyte E:\caffetest\mnist\mnist\mnist_test_lmdb

pause

保存后,双击运行,生成了两个文件夹:mnist_train_lmdb和mnist_test_lmdb

 2.打开lenet_train_test.prototxt

然后打开lenet_solver.prototxt,修改第二行即可,

可以将prototxt文件放在以下的网址,查看网络的结构:http://ethereon.github.io/netscope/#/editor

3.使用caffe.exe 训练网络

利用命令行cd到caffe.exe的工作目录,输入以下的命令:

caffe train -solver=lenet_solver.prototxt -gpu 0

可以得到以下的训练结果:

4.使用Python调用caffe

环境:python+Anaconda;同时将编译生成的python文件夹放在..\anaconda2\Lib\site-packages中。

import caffecaffe.set_mode_gpu()solver = caffe.SGDSolver('lenet_solver.prototxt')solver.solve()

运行代码可以得到以下的结果:

 

转载于:https://www.cnblogs.com/youngsea/p/9430875.html

你可能感兴趣的文章
JSP 动作 JavaBean
查看>>
搜索系统搭建 mysolr
查看>>
xcode 安装插件管理工具Alcatraz
查看>>
iOS10适配远程推送
查看>>
华三模拟器配置Telnet
查看>>
Python 6 个字典操作你必须知道
查看>>
Get智能谈AI写作:痛则思变,让人工智能成为你新媒体的写作利器
查看>>
全新lenovo System x3650M配置raid0,raid1,raid10
查看>>
Openstack Dashboard 二次开发
查看>>
linux 基础命令大全(二)
查看>>
CSS美化超链接
查看>>
Java23种设计模式案例:建造者模式(builder)
查看>>
docker挂载本地目录,实现文件共享
查看>>
再有人问你分布式锁,就把这个丢给他!
查看>>
Haproxy、Keepalived双主高可用负载均衡
查看>>
【No.5 Ionic】修改 应用名,icon,启动界面
查看>>
我的友情链接
查看>>
搭建高可用mongodb集群(一)——配置mongodb
查看>>
文件操作
查看>>
keepalive之mysql故障自动切换脚本
查看>>