首页 > 其他 > 详细

Caffe训练cafir10

时间:2017-01-22 11:56:37      阅读:1164      评论:0      收藏:0      [点我收藏+]

Caffe训练cafir10

安装Cygwin

可以执行.sh脚本

  1. cd caffe-master/data/cifar10
  2. get_cifar10

下载下来的文件列表

batches.meta.txt

data_batch_1.bin

data_batch_2.bin

data_batch_3.bin

data_batch_4.bin

data_batch_5.bin

test_batch.bin

caffe-master\Build\x64\Release目录下convert_cifar_data.exe

  1. convert_cifar_data input output lmdb

在output文件加下生成两个文件夹

cifar10_test_lmdb

cifar10_train_lmdb

拷贝到caffe-master\examples\cifar10目录下。

接下来与caffe训练mnist一样。

有三个网络模型可以选择,参数不同。

cifar10_full.prototxt

cifar10_full_sigmoid_solver.prototxt

cifar10_quick.prototxt

 

生成均值文件

  1. compute_image_mean cifar10_train_lmdb mean.binaryproto

创建标签文件

synset_words.txt

0airplane

1automobile

2bird

3cat

4deer

5dog

6frog

7horse

8ship

9truck

修改cifar10_quick_solver.prototxt与cifar10_quick_train_test.prototxt文件中的相对路径为绝对路径。

训练

  1. caffe train –solver=cifar10_quick_solver.prototxt

quick网的训练结果

技术分享

full_sigmoid网的训练结果

技术分享

 

利用网络对单幅图像进行分类

  1. Classification cifar10_quick.prototxt cifar10_quick_iter_4000.caffemodel.h5 mean.binaryproto synet_words.txt "..\images\cat.jpg"

技术分享

 

 

 

 

训练自己的数据

技术分享

 

技术分享

Caffe训练cafir10

原文:http://www.cnblogs.com/yizhichun/p/6339802.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!