Yolo v5

Yolo v5

学习笔记--Yolo v5

• 环境配置

---

环境配置

1. Ubuntu20.0.4下安装cuda11.0和cudnn
2. 基于Ubuntu20.04安装Nvidia驱动
3. Ubuntu20.04安装opencv

本来想环境配置完用GPU跑一下Yolov4的，但是Yolov4的检测图片上没有结果显示，可能是cuda等配置的版本问题，暂时还没有找到解决方法（CPU跑是可以的，只需要把 Makefile 里的 GPU、CUDNN、CUDNN_HALF都设置为0，但是速度会很慢）；就先试着跑了一下Yolov5，参考，想仅实时检测人的话，使用这句命令python detect.py --source 0 --classes 0