zoukankan      html  css  js  c++  java
  • 2020/5/10

    安装和使用YOLOv4

    darknet
    YOLOv4 是使用和YOLOv3差不多,主要记录不同点。

    Makefile

    YOLOv4 的Makefile前几行是这样的。

      
    GPU=0
    CUDNN=0
    CUDNN_HALF=0 # new
    OPENCV=0
    AVX=0 # new
    OPENMP=0
    LIBSO=0
    ZED_CAMERA=0 # ZED SDK 3.0 and above
    ZED_CAMERA_v2_8=0 # ZED SDK 2.X
    

    同样的, 要使用GPU 另GPU=1, CUDNN=1。

    值得注意的是OPENCV 这个,在YOLOv3中不是用它的话编译不会出错,但是YOLOv4中却会出错,解决办法。
    将.cfg文件中的mosaic改为0

    learning_rate=0.00261
    burn_in=1000
    max_batches = 6000
    policy=steps
    steps=1600,1800
    scales=.1,.1
    
    #cutmix=1
    mosaic=0 # 这里改0
    

    如果出现段错误,多半是batchsize开的太大,减小这个值。

    适用于自己的模型

    我用的是单目标检测,安装下面这个指导,我完成参数的修改, 结合gayhub上的指导一起看。
    窗口环境下Yolo V4的目标检测算法

    尤其要注意的是.cfg文件参数的修改。

    Python 脚本的编写

    以前偷懒没怎么自己写Python脚本,今天将dlib 的 xml 文件转化成YOLO格式的txt文件,并且生成YOLO的文件树结构,累死我了。。

    ├─JPEGImages # 图片位置
    ├─labels # 标记
    ├─weights # 保存权重
    ├─test # 测试.txt
    └─train # 训练.txt
    

    本来想用regex, 正则表达式真不是人看的东西。。。。

    明天还有写, 明天再总结吧。

    还有早课,赶紧睡!

  • 相关阅读:
    Ubuntu中pip的疑难杂症
    Python 分类方法记录
    Python 绘制热图
    脚本学习
    Ubuntu 基础使用教程
    机器学习各种相似性度量及Python实现
    Ubuntu 16.04安装R及Rstudio
    机器学习和数据挖掘领域大牛
    vux+vue-cli3.0坑
    函数的抖动以及节流
  • 原文地址:https://www.cnblogs.com/hichens/p/12866493.html
Copyright © 2011-2022 走看看