修改YOLO-tiny结构之后(二)

1.修改完之后的Tiny网络有49层(23+26),检测尺度仍然是2626,1313;

2.训练:
1)这次将max_batches设置为80200;学习率在55000,65000时降低为原来的0.1;
还是使用VOC2007,2012数据集进行训练,将2007_test作为模型验证集。

2)训练完之后得到所有的权重文件my_yololite_final.weights等。
ps:若想在训练之后将训练过程中的IoU,mAP,recall等用图形可视化,可以在训练时利用类似下面的命令保存log文件:

./darknet detector train cfg/voc.data cfg/yolov3-tiny.cfg yolov3-tiny.conv.15  -gpus 0,1  2>1 | tee visualization/tiny_yolov3.log 

注意需要将命令中的参数改成自己训练的文件。
可视化可以参考YOLO训练中的可视化

3.利用训练好的模型进行测试
1)验证集下的recall:
可以直接调用Darknet框架下的命令:

./darknet detector recall cfg/voc.data cfg/yolov3-tiny-test.cfg backup/yolov3-tiny_70000.weights

输出结果:
在这里插入图片描述2)计算mAP
主要参考YOLO中map值的计算
先进行验证生成results/*.txt

./darknet detector valid cfg/voc.data cfg/yolov3-tiny-test.cfg backup/yolov3-tiny_70000.weights -out "" -gpu 0 -thresh .5

然后利用voc_eval.py下载
新建compute_mAP.py文件测试即可出结果:
在这里插入图片描述初学者暂时学到这了,有机会之后再更新。

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值