YOLO v3曲线可视化

第一,在进行训练时进行日志保存,保存在darknet目录下。代码如下:

保证在训练过程中生成train_yolov3.log文件,方便后续进行使用(注意:本人是在python2环境下运行)

第二,使用extract_log.py格式化Log文件,代码如下

链接:https://pan.baidu.com/s/1ksEy037rU6MhpmPBnzEqjg 
提取码:h8lv

 

运行后,log文件解析出来两个.txt文件(train_log_loss.txt; train_log_iou.txt)

第三,使用train_loss_visualization.py脚本绘制loss曲线,代码如下

链接:https://pan.baidu.com/s/1at1QJzeCjePNDpbdVhudTQ 
提取码:x16f

第四,使用train_iou_visualization.py可视化iou曲线,代码如下:

链接:https://pan.baidu.com/s/1SfFw54dElk7wo4ySCqCpjQ 
提取码:yack

第五,mAP的计算

           1.使用代码生成检测结果文件:

./darknet detector valid cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_final.weights -out person.txt -gpu 0 -thresh .5

(这里person.txt是我的训练分类,我只训练了person 这一类)

           2.将faster-rcnn/lib/datasets/voc_eval.py考到darknet目录下,之后在自己新建一个compute_mAP.py文件(如果没有voc_eval.py,下边我提供了百度云,ke'y可以直接去下载)

链接:https://pan.baidu.com/s/1Hg7ibv_Wqnu_IKE2TwqWiw 
提取码:0ij3(voc_eval.py)

链接:https://pan.baidu.com/s/1kqLQ3sjLncA3eHwOWsH3-A 
提取码:sweu(compute_mAP.py)

compute_mAP.py 根据自己的路径进行修改即可

          3.运行compute_mAP.py即可

  • 0
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 9
    评论
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值