在caffe做数据

我不知道怎么做数据,我也看了官方教程,感觉不全
已邀请:

breeze - 90后小学生

赞同来自: yanhe 刘昕_ICT 李扬

1、将自己的图片按照imegenet的格式弄好:train文件夹中,每个类别的文件单独放一个文件夹,例如 one、two文件夹中分别存放不同的图片one1.jpg到onen.jpg、two1.jpg到twon.jpg,在train.txt中,生成标签,格式如下:one/one1.jpg 0
       one/one2.jpg 0……
命名规则为,文件名(包含路径)+空格+label(label从0开始,并且要连续,否则会报错)。 val中不需要加路径,后面标签和train一致。                                                                     
caffe中默认的类别从0开始,val文件夹,所有的图片放到文件夹中,命名格式和和train文件夹中一样。

2、在caffe-master/data下新建文件夹mydata,然后将第一步生成的两个文件夹和两个txt文件放到caffe-master/data/mydata(caffe-master为caffe的根目录,mydata自己命名)。

3、进入caffe-master/example/imagenet,修改create_imagenet.sh。修改内容如下:
DATA=data/mydata 表示存放图片的地方
TRAIN_DATA_ROOT=data/mydata/train/
VAL_DATA_ROOT=data/mydata/val/
RESIZE=true
if $RESIZE; then
  RESIZE_HEIGHT=256 //自行修改大小
  RESIZE_WIDTH=256
else
  RESIZE_HEIGHT=0
  RESIZE_WIDTH=0
自行将图片resize成相应尺寸。
注、:如果是灰度图,在resize_width=$RESIZE_WIDTH \下面加上一行代码gray=true \就可以了。
 

uestc_yang - UESTC

赞同来自:

什么叫做数据,是将图片存在leveldb/lmdb??

yanhe

赞同来自:

我想训练一个自己的模型,是按照这个链接来的http://caffecn.cn/%3F/question ... er_91,图片都是从imgaeNet上下载的。如果我用自己的图片训练一个模型,应该怎样准备我的数据?

要回复问题请先登录注册