https://github.com/ultralytics/yolov5/
1、在Anaconda Promptx新建conda虚拟环境
conda create -n py39_yolov5 python=3.9
2、激活虚拟环境
conda activate py39_yolov5
3、虚拟环境下装所需依赖
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cpuonly -c pytorch ---------这里使用cpu
# CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1
cudatoolkit=11.6 -c pytorch-------有英伟达显卡的话可以用这个
4、下载yolov5和模型文件,并将模型文件放入yolov5下载的目录里
进入到yolov5目录并安装依赖pip install -r requirements.txt
安装时遇到编码问题
修改编码也没用,后面通过升级conda( conda update -n base conda)后重新安装pip install解决
安装过程因网络问题较慢,需等待较久
5、验证测试yolov5环境是否安装ok
在runs\detect\exp目录下有检测的结果图片
6、测试下训练
python train.py
训练中出错
按如下修改后重新train
cpu训练很慢,pc会很卡
训练结果会在runs\train\exp生成