另有Yolov4模型的各种部署方式,详见github。
0.配置环境
此项目使用的训练环境:Ubuntu 18.04.5, CUDA Version: 10.1, Darknet A版, Nvidia 2080 ti 显卡.
其中 Darknet 的配置参考这里
1.数据准备
这个项目使用的反光衣检测数据集来自reflective-clothes-detect and dataset yolov5,
共有反光衣检测数据1083张。
- 进入百度云盘下载链接 提取码->(dooh).
- 将数据文件夹中‘labels/'下的文件复制到‘JPEGImages/’中,保证两者在同一个文件夹下。