1.安装Anaconda

Anaconda包括 conda、python以及一大堆安装好的工具包,比如:numpy、pandas等,支持Linux,Mac,Windows,包含了众多流行的科学计算、数据分析的 Python 包。

https://www.anaconda.com/download/success

2.安装pytorch

(1)终端创建虚拟环境(python39_pytorch为自定义的环境名),并安装python3.9

conda create -n python39_pytorch python=3.9 # 创建环境

conda activate python39_pytorch # 激活环境

(2)打开 https://pytorch.org/ 复制命令到终端下载

pip3 install torch torchvision torchaudio

(3)安装 yolov5 依赖


解压后,进入目录

cd yolov5-master

pip install requirements.txt

测试:

python detect.py --source ./data/images/bus.jpg --weights weights/yolov5s.pt --conf 0.4

可能会遇到权重文件yolov5s.pt下载失败,直接去官网下载https://github.com/ultralytics/yolov5/releases/download/v7.0/yolov5s.pt ,将文件保存到项目的weights/目录下,确保路径与命令一致。

测试完成后,进入/yolov5-master/runs/detect/exp2 文件夹,出现以下图片 表示模型部署完毕