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

你可能感兴趣的:(Yolo,其他)