您当前的位置:首页 > 计算机 > 编程开发 > 人工智能

YOLOv4配置与训练

时间:02-02来源:作者:点击数:

环境配置

  • cuda10.2
  • cudnn8.2.0
  • opencv 3.4.10

darknet配置

官网下载darknet, 按照官方步骤Install

注意:

  • 记得在Makefile中使用GPU
  • 如果是通道为1的灰度图像,需要修改源文件和cfg文件,见文章

准备数据集并训练

参考文章

注意:

  • 拷贝的darknet可执行文件是和编译环境有关的,必须是本机编译出来的darknet在本机使用。

示例训练代码

见实验室仓库myboyhood yolov4

训练文件下载百度云秘钥: trdv

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门