在实习中,有一个任务是训练出一个车辆检测的yolov4模型。感觉yolov4的功能好强大啊!特此记录一下自己在使用过程中的一些改动。
首先肯定是要配置相关的环境:OpenCV,Cmake,还有cuda,cudnn这些就不多说了。配置环境基本项,多找博客多试一试。
我用的系统环境是:
deepin系统, OpenCV3.4.16, Cmake3.22.0, cuda11.3, cudnn8.2.1。
配置好环境,下载了源码肯定动手开始训练,源码很多。需要修改的我梳理了一下:
1.预训练权重
首先下载好预训练权重放在darknet文件夹中(建议在里面新建一个项目文件夹例如xingding,把所有的文件都放在里面)。
2.复制配置文件
复制cfg文件夹里面的yolov4-custom.cfg文件。然后修改里面的内容:
max_batches = classes*2000
steps = 0.8*max_batches,0.9*max_batches
ctrl+F 搜索[