Lovely-Pig / npucar-yolov4

npucar-yolov4

概述

讯飞智能车图像识别-yolov4

运行方法

本项目比较耗GPU，所以租用了openbayes的GPU资源进行训练

1. 准备工作

   安装相关依赖 + 下载预训练模型

   $ sh prework.sh

2. 使用自己的数据集训练模型

   $ python train.py

3. 执行检测任务

   # 对文件夹下的图片进行检测
   $ python detect.py
   
   # 摄像头实时检测
   $ python cam_detect.py

参考资料

• PyTorch-YOLOv3
• yolo
• yolov3-paper
• openbayes