在VOC上训练YOLO2

1、获取Pascal VOC Data并解压

wget https://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar
wget https://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar
wget https://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar
tar xf VOCtrainval_11-May-2012.tar
tar xf VOCtrainval_06-Nov-2007.tar
tar xf VOCtest_06-Nov-2007.tar

2、生成VOC标签
现在我们需要生成Darknet使用的标签文件,要生成这些文件,我们将voc_label.py在Darknet的scripts/目录中运行脚本。

python voc_label.py

或者自己下载再运行

wget https://pjreddie.com/media/files/voc_label.py
python voc_label.py

3、运行脚本后,在该目录下生成几个文件,表示运行成功

2007_test.txt  
2007_train.txt  
2007_val.txt    
2012_train.txt 
2012_val.txt    

4、修改Pascal Data的Cfg
修改/darknet/cfg/voc.data配置文件来指向你自己的数据

 classes= 20
 train  = <path-to-voc>/train.txt
 valid  = <path-to-voc>2007_test.txt
 names = data/voc.names
 backup = /backup

注意:1、backup文件若是不存在需要自己手动创建一个。
5、下载预训练的权重

wget https://pjreddie.com/media/files/darknet19_448.conv.23

6、开始训练
注意:路径要对应(darknet19_448.conv.23)放在cfg同一级目录。

./darknet detector train cfg/voc.data cfg/yolo-voc.cfg darknet19_448.conv.23

7、测试训练模型的效果

cd darknet
./darknet detector test cfg/voc.data cfg/yolo-voc.cfg 你的backup目录/my_final.weights data/测试图片.jpg -thresh 0.05 (可调整设置阈值)

其它数据集如在COCO上训练原理一样。
参考博客:https://pjreddie.com/darknet/yolo/

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值