使用 golang 标准库不到 200 行代码实现 yolov4 项目脚手架命令行工具

原创

于 2022-02-11 11:03:10 发布 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

知识因分享而具有价值，回忆昨天，把握今天，期待明天。 —— 驭鲸·环球

文章标签：

#golang #开发语言 #后端

DarknetConfig是基于Golang的Darknet神经网络配置文件生成器，简化创建项目流程。通过调整*.cfg配置文件中的网络分辨率能提升对象检测精度，但会增加GPU负载。确保数据集准确无误，负样本数量与正样本相当，并正确标记对象以优化检测效果。项目源码可在Gitee找到。

DarknetConfig

DarknetConfig 是一个创建 darknet 神经网络配置文件项目的便捷工具。

Usage of DarknetConfig.exe:
  -batch int
        batch default value 64 (default 3)
  -classes int
        classes default value 3 (default 3)
  -height int
        height default value 608 (default 608)
  -path string
        project path (default "./temp")
  -subdivision int
        subdivision default value 16 (default 3)
  -type int
        type enum [0:yolov4,1:yolov4-tiny] default value 0
  -width int
        width default value 608 (default 608)

使用 DarknetConfig.exe -path ./test 命令创建 test 项目。

项目结构如下：

test
│  readme.txt
│  yolov4.cfg       //yolov4 网络配置,默认内容已根据命令行参数更改，这里不需要做更改。
│
├─backup            //存放训练后的模型
└─data
    │  obj.data     //存放数据配置
    │  obj.names    //存放标签
    │  train.txt    //存放训练集图片路径
    │  valid.txt    //未使用
    │
    └─obj           //存放正负样本图像资源