Caffe:制作与训练自己的图片数据

来源:互联网 发布:显卡更新驱动负优化 编辑:程序博客网 时间:2024/05/16 19:16

一、制作数据

根据上篇内容来制作数据。
我用的是中草药的数据,包括三种药材;
cheqianzi:700张
JianLianHua:1000张
pugongying:500张
其中的百分之二十用于test,百分之八十用于train。
test:140+200+100=440张
train:560+800+400=1760张
文件存放如下:
这里写图片描述

在data/herbal_demo/下建立两个文件夹:
这里写图片描述
分别存放test和train图片。

二、制作数据

在examples/herbal_demo/下建立create_filelist.sh文件
内容如下:

#!/usr/bin/env shDATA=data/herbal/MY=examples/herbal_demoecho "Create train.txt..."rm -rf $MY/train.txtfind $DATA/train -name cq2015*.jpg | cut -d '/' -f4-5 | sed "s/$/ 1/">>$MY/train.txtfind $DATA/train -name JLHZZ*.JPG | cut -d '/' -f4-5 | sed "s/$/ 2/">>$MY/train.txtfind $DATA/train -name py201512*.jpg | cut -d '/' -f4-5 | sed "s/$/ 3/">>$MY/train.txtecho "Create test.txt..."rm -rf $MY/test.txtfind $DATA/train -name cq2015*.jpg | cut -d '/' -f4-5 | sed "s/$/ 1/">>$MY/test.txtfind $DATA/train -name JLHZZ*.JPG | cut -d '/' -f4-5 | sed "s/$/ 2/">>$MY/test.txtfind $DATA/train -name py201512*.jpg | cut -d '/' -f4-5 | sed "s/$/ 3/">>$MY/test.txtecho "All done"

$ ./examples/herbal_demo/create_filelist.sh
运行后:
生成test.txt和train.txt文件。
这里写图片描述
待续。。

0 0
原创粉丝点击