运行环境
环境: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文件夹可以看到处理的图片。