训练模型

1. 数据准备

  1.1 原始数据的准备

    jpg文件等  

  1.2 辅助数据的准备

    生产train.txt和val.txt

    参考代码:

# /usr/bin/env sh
DATA=examples/images
echo "Create train.txt..."
rm -rf $DATA/train.txt
find $DATA -name *cat.jpg | cut -d '/' -f3 | sed "s/$/ 1/">>$DATA/train.txt
find $DATA -name *bike.jpg | cut -d '/' -f3 | sed "s/$/ 2/">>$DATA/tmp.txt
cat $DATA/tmp.txt>>$DATA/train.txt
rm -rf $DATA/tmp.txt
echo "Done.."

    针对自己的数据,在terminal中直接操作:

      find airport_inside -name *.jpg | cut -d '/' -f2-3 | sed "s/$/ 0/">>train.txt

      find artstudio -name *.jpg | cut -d '/' -f2-3 | sed "s/$/ 1/">>train.txt

      ... ...

      cat tmp.txt>>train.txt

    

2. 计算图片平均值

3. 模型定义

4. 训练网络

5. 恢复训练

6. 测试

 http://nbviewer.jupyter.org/github/BVLC/caffe/blob/master/examples/00-classification.ipynb

http://blog.csdn.net/hjimce/article/details/49248231

 

posted @ 2016-09-05 19:56  balmy233  阅读(171)  评论(0编辑  收藏  举报