博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Ubuntu+caffe训练cifar-10数据集
阅读量:4701 次
发布时间:2019-06-09

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

1. 下载cifar-10

ciffar-10数据集包含10种物体分类,50000张训练图片,10000张图片。

在终端执行指令下载cifar-10数据集(二进制文件):

[html]   
 
  1. cd ~/caffe-master  
  2. ./data/cifar10/get_cifar10.sh  

在./data/cifar10文件夹下生成5个.bin的训练数据集合1个测试数据集:

2. 生成lmdb以及均值文件

[html]   
 
  1. ./examples/cifar10/create_cifar10.sh  

执行之后在./examples/cifar10下生成训练和测试lmdb文件和均值mean.binaryproto文件

3. 根据系统修改运行模式(CPU或GPU)

[html]   
 
  1. sudo gedit ./examples/cifar10/cifar10_quick_solver.prototxt  

修改最后一行 solver_mode 的配置为CPU或GPU

4. 执行训练

执行cifar10文件夹下的train_quick.sh开始训练:

[html]   
 
  1. ./examples/cifar10/train_quick.sh  

4000次的迭代在CPU上训练耗时约30min,最终精度是84%:

转载于:https://www.cnblogs.com/mtcnn/p/9411828.html

你可能感兴趣的文章
从零开始人工智能AI(一)-k-nn-矩阵
查看>>
抽奖概率计算
查看>>
x64的调用约定
查看>>
linux 禁ping
查看>>
设计一个栈,除了pop与push方法,还支持Min方法,可返回栈元素中的最小值,push、pop、min三个方法的时间复杂度必须是O(1)...
查看>>
函数定义与使用,字符串类型
查看>>
Java - TreeMap源码解析 + 红黑树
查看>>
Android-事件分发(ViewGroup)
查看>>
Djngo 请求的生命周期
查看>>
[kuangbin带你飞]专题1-23题目清单总结
查看>>
EF6.0批量插入
查看>>
Knockout v3.4.0 中文版教程-1-入门和安装
查看>>
JavaScript 基础一
查看>>
数据库范式1NF 2NF 3NF BCNF
查看>>
ComboBox控件隐藏fieldLabel不能隐藏问题解决
查看>>
Hibernate自动生成DO手写DAO的注意事项
查看>>
IDEA中打包Spark项目提示Error:(16, 48) java: -source 1.5 中不支持 lambda 表达式
查看>>
poj 3013 Big Christmas Tree 最短路 dijkstra算法
查看>>
Java 多线程,Thread,Runnable,Callable
查看>>
HTTP长连接、短连接究竟是什么?
查看>>