yolo v2微调

论文:YOLO9000: Better, Faster, Stronger
参考论文笔记:
http://blog.csdn.net/jesse_mx/article/details/53925356

1.准备数据,标签做成如下格式:
object class x y width height
如果下载的数据标签格式是voc,可以下载python文件转换:
curl -O http://pjreddie.com/media/files/voc_label.py

2.修改cfg文件
参考:http://blog.csdn.net/u014540717/article/details/53424843
如果你想用22层模型的就修改cfg/yolo-voc.cfg,你想用9层的模型就修改cfg/tiny-yolo-voc.cfg,两者修改方式一样,我们以yolo-voc.cfg为例:
复制cfg文件

$cp cfg/yolo-voc.cfg cfg/yolo-fddb.cfg
打开yolo-fddb.cfg文件,并作如下修改

a. 将learning_rate=0.0001改为learning_rate=0.00005
b. 将max_batches = 45000改为max_batches = 200000
c. 将classes=20改为classes=1
d. 将最后一层[convolutional]层的filters=125改为filters=30,filters的计算公式如下,请根据你自己数据的类别数量修改
filters=num∗(classes+coords+1)=5∗(1+4+1)=30

3.修改voc.names
微调几类,保存几类,我们只微调一类car

4.修改voc.dat文件,指定训练数据,验证数据,类别名称,结果模型保存位置
classes= 20
train = /train.txt
valid = 2007_test.txt
names = data/voc.names
backup = backup

5.下载预训练卷积权重
curl -O http://pjreddie.com/media/files/darknet19_448.conv.23
或者自己生成预训练的卷积权值
./darknet partial cfg/darknet19_448.cfg darknet19_448.weights darknet19_448.conv.23 23

6.开始微调
./darknet detector train cfg/voc.data cfg/yolo-voc.cfg darknet19_448.conv.23

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值