yoloV2训练

yolov2:https://pjreddie.com/darknet/yolov2
下载darknet
git clone https://github.com/pjreddie/darknet
cd darknet
make

下载yolov2的参数文件.weights
wget https://pjreddie.com/media/files/yolov2.weights

测试对应的模型文件
./darknet detect cfg/yolov2.cfg yolov2.weights data/dog.jpg

通过某个数据集来测试对应模型文件
./darknet detector test cfg/coco.data cfg/yolov2.cfg yolov2.weights data/dog.jpg

自己训练一个基于voc数据集的yolo模型

准备数据:
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
...

下载别人标签好的文件(或者自己去标注)
wget https://pjreddie.com/media/files/voc_label.py

python voc_label.py 会生成几个txt文件,对应的数据文件

cat 2007_train.txt 2007_val.txt 2012_*.txt > train.txt

将那几个txt文件cat到一个train.txt文件中用于训练过程

修改voc.data,自己的类别有多少就对应修改相关参数及文件路径
为卷积层准备预训练参数文件
wget https://pjreddie.com/media/files/darknet19_448.conv.23

修改Makefile文件里的相关参数
训练模型:
./darknet detector train cfg/voc.data cfg/yolov2-voc.cfg darknet19_448.conv.23

模型训练完成后保存的文件在backup文件夹中
调用训练后的参数文件对模型进行测试:视频&图片

detect test:video&picture with my trainning weights file through the cfg file
./darknet detector demo cfg/voc.data cfg/yolov2-voc.cfg backup/yolov2-voc_final.weights test2.mp4


./darknet detect cfg/yolov2-voc.cfg backup/yolov2-voc_final.weights data/dog.jpg
 


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值