ubuntu16.04运行YOLOV5并部署

运行环境

环境：ubuntu 16.04

CUDA: 10.2

执行模型推理

1. 下载yoloV5 repo

git clone https://github.com/ultralytics/yolov5

2. 用conda新建python环境

conda create -n yolo python=3.8

3.  安装需求包

pip install -r requirements.txt

4. (requremets中包含pytorch的安装，但是实际操作中我又安装了一次)

 conda install pytorch torchivsion -c pytorch

5, 运行预编译模型,会自动下载模型，生成yolov5s.pt,并处理图片。

python detect.py

6. 在runs/detect/expx文件夹可以看到处理的图片。

模型量化