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

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           //存放正负样本图像资源

如何提高对象检测率？

1. 在 *.cfg 配置文件中使用更大的网络分辨率，它会提高精度。同时它会加大 GPU 的负担。
2. 清洗数据集，数据集中不应该有任何粗错误的标签。
3. 负样本（空的 .txt 文件）数量控制在和正样本数量一样多。
4. 标记对象的最佳方法，仅标记对象可见部分，或标记比整体对象多一点（有一点点间隙），或标记对象的可见和重叠部分。

仓库地址：https://gitee.com/yujinghuan