darknet 可视化yolo损失曲线

9人阅读 评论(0) 收藏 举报
分类:

保存日志命令:

/darknet detector train my/mot/person.data my/mot/densenet201_yolo.cfg backup/mot/densenet201_yolo_final.weights >> log/mot-ramdon.log

python可视化代码:

#plot.py
import argparse 
import sys
import matplotlib.pyplot as plt
def main(argv):
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-file",
        help = "path to log file"
        )
    args = parser.parse_args()
    f = open(args.file)
    
    lines  = [line.rstrip("\n") for line in f.readlines()]
    
    numbers = {'1','2','3','4','5','6','7','8','9'}
    iters = []
    loss = []
    
    fig,ax = plt.subplots()
    prev_line = ""
    for line in lines:
        args = line.split(' ')
        if args[0][-1:]==':' and args[0][0] in numbers :
            iters.append(int(args[0][:-1]))            
            loss.append(float(args[2]))
             
    ax.plot(iters,loss)
    plt.xlabel('iters')
    plt.ylabel('loss')
    plt.grid()
    ticks = range(0,250,10)
    
    #ax.set_yticks(ticks)
    plt.show()
    
if __name__ == "__main__":
    main(sys.argv)

效果图:

源地址:https://ricky.moe/2017/11/04/yolo-training-visualization/

查看评论

【Darknet】【yolo v2】训练自己数据集的一些心得----VOC格式

-------【2017.11.2更新】------------SSD传送门----------http://blog.csdn.net/renhanchi/article/details/78411...
  • renhanchi
  • renhanchi
  • 2017-05-02 10:02:34
  • 17426

yolo v2 损失函数源码解读

前提说明:     1, 关于 yolo 和 yolo v2 的详细解释请移步至如下两个链接,或者直接看论文(我自己有想写,但思前想后下面两个链接中的文章质量实在是太好了_(:з」∠)_)      ...
  • xueyingxue001
  • xueyingxue001
  • 2017-06-01 15:26:12
  • 5891

YOLO模型训练可视化训练过程中的中间参数

等待训练结束后(有时还没等结束模型就开始发散了),因此需要检测各项指标(如loss)是否达到了我们期望的数值,如果没有,要分析为什么。可视化训练过程的中间参数可以帮助我们分析问题。 可视化中间参数需...
  • yudiemiaomiao
  • yudiemiaomiao
  • 2017-05-18 15:06:39
  • 3353

【YOLO】详解:YOLO-darknet训练自己的数据

准备训练数据 使用darknet训练自己的YOLO模型需要将数据转成darknet需要的格式,每张图片对应一个.txt的label文件,文件格式如下: object-class是类的索引...
  • Jinlong_Xu
  • Jinlong_Xu
  • 2017-07-20 20:10:22
  • 6059

YOLO源码详解(三)- 前向传播(forward)

本系列作者:木凌 时间:2016年11月。 文章连接:http://blog.csdn.net/u014540717 QQ交流群:554590241 一、主函数void forward_netwo...
  • u014540717
  • u014540717
  • 2016-11-19 17:55:31
  • 10950

基于darknet框架的mobilenet

这里介绍关于Google的文章(MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications)中...
  • muwu5635
  • muwu5635
  • 2017-06-28 09:56:20
  • 4918

yolo v2 损失函数源码(训练核心代码)解读和其实现原理

前提说明:     1, 关于 yolo 和 yolo v2 的详细解释请移步至如下两个链接,或者直接看论文(我自己有想写 yolo 的教程,但思前想后下面两个链接中的文章质量实在是太好了_(:з」...
  • yangdashi888
  • yangdashi888
  • 2017-09-19 09:29:37
  • 1782

yolo v2的android移植

  • 2017年05月12日 16:11
  • 2.39MB
  • 下载

详解:YOLO-darknet训练自己的数据

转自:http://blog.csdn.net/Jinlong_Xu/article/details/75577007 准备训练数据 使用darknet训练自己的YOLO模型需要将...
  • yaningli
  • yaningli
  • 2017-09-25 17:26:17
  • 800

Linux、Windows下试用DarkNet之YoLo CPU物体识别

关于YOLO: YOLO——You Only Look Once Faster RCNN需要对20k个anchor box进行判断是否是物体,然后再进行物体识别,分成了两步。  YOLO(You...
  • houwenbin1986
  • houwenbin1986
  • 2017-01-10 12:56:45
  • 5343
    个人资料
    等级:
    访问量: 169
    积分: 15
    排名: 206万+
    文章分类
    文章存档