caffe训练自己的模型步骤

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
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值