caffe训练自己的模型步骤

来源:互联网 发布:北京seo哪家好 编辑:程序博客网 时间:2024/05/01 03:18

1.准备数据

创建目录mkdir mydata,mkdir mydata/train,mkdir mydata/val,在train下放训练数据,在val下放验证数据。

生成train.txt、val.txt、test.txt


1.1生成val.txt:find  -name *.png |grep -v train | cut -d/ -f3>val.txt

1.2生成train.txt:find  -name *.png |grep  train | cut -d/ -f3-4 > train.txt

添加分类:注意分类和值之间只能添加一个空格,否则执行create_imagenet.sh会报类似如下错误:

E0818 16:36:51.530107  9428 io.cpp:80] Could not open or find file /home/caffe/caffe-master/.../train_zc_1072.png 

1.3添加分类

sed -i '1,1200s/.*/& 0/' train.txt和sed -i'1201,2400s/.*/& 1/' train.txt。

val类似,但是val以为是两类,所以我一开始放到了2个文件,采用生成val.txt的方法,生成两个文件以后再合并,再添加分类。

也可以添加分类后再合并,合并如下:

vi 文件1,文件2,复制命令:1,&y,然后:bp来转换到下一个文件,然后P,就会粘贴到光标处了。

0 0
原创粉丝点击