yolov5项目拉取及环境安装配置
前言
本文主要用来记录拉取yolov5项目。
一、创建conda 虚拟环境
首先创建一个新的conda虚拟环境。conda create yolov
安装符合pytorch+torchvision，命令如下所示：
首先在官网查看需要下载的版本命令，类似下面这条命令，打开 Anaconda Prompt 进入虚拟环境，conda activate yolov，执行下面的命令等待安装即可
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 -c pytorch
二、拉取github项目
1、pycharm打开项目，配置环境
命令如下：
pip install - r requirements.txt
如果遇到下面得报错：
There was a problem confirming the ssl certificate: HTTPSConnectio
nPool(host='pypi.tuna.tsinghua.edu.cn', port=443):Max retries exceeded with url: /simple/numpy/ (Caused by SSLError(SSLEOFError(8, 'EOF occurred in violation of protocol (_ssl.c:1131)'))) - skipping
就使用下面的命令：换源即可
pip install - r requirements.txt -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
安装成功命令提示：
Attempting uninstall: numpyFound existing installation: numpy 1.16.6Uninstalling numpy-1.16.6:Successfully uninstalled numpy-1.16.6
Successfully installed MarkupSafe-2.1.2 PyYAML-6.0 absl-py-1.4.0 cachetools-5.3.0 charset-normalizer-3.1.0 colorama-0.4.6 google-auth-2
.17.2 google-auth-oauthlib-0.4.6 grpcio-1.51.3 idna-3.4 importlib-metadata-6.1.0 markdown-3.4.3 numpy-1.24.2 oauthlib-3.2.2 opencv-pyth
on-4.7.0.72 pandas-1.5.3 protobuf-4.22.1 pyasn1-0.4.8 pyasn1-modules-0.2.8 pycocotools-2.0.6 pytz-2023.3 requests-2.28.2 requests-oauth
lib-1.3.1 rsa-4.9 scipy-1.10.1 seaborn-0.12.2 tensorboard-2.12.0 tensorboard-data-server-0.7.0 tensorboard-plugin-wit-1.8.1 thop-0.1.1.post2209072238 tqdm-4.65.0 urllib3-1.26.15 werkzeug-2.2.3
当出现上面的信息，说明环境安装成功。
三、bug解决
1、运行 detect.py，遇到报错
报错内容如下所示：
DLL load failed while importing _imaging: 找不到指定的模块
解决办法：报错的代码行位置是 from . import _imaging as core
可以升级 pillow模块，命令如下：pip install -U Pillow
2、遇到报错2
报错内容如下所示：
AttributeError: Can‘t get attribute ‘SPPF‘ on ＜module ‘models.common‘ from ...
报错情况说明：该报错是指 models目录下的 common.py 文件找不到 SPPF 函数。
解决办法：就是查看新版本的 yolov6及以上，把这个代码拷贝复制到该文件内即可。
3、遇到报错3
报错The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3
报错原因分析，指的是神经网络某个层，输入的张量维度对不上。
解决办法，暂无。
开个玩笑，办法肯定有的。而且出现报错2和报错3的原因是同一个。
原因分析，因为我的 yolov5s.pt 模型是手动去官网下载的，主要原因就是，该模型没有从正确的地方下载该模型 。
正确下载模型链接
https://github.com/ultralytics/yolov5/releases/tag/v5.0
原因说明：就是你选的是v5版本的代码，模型应该也是v5版本的，而不是去其他版本页面下载v5模型，因为两个文件内的某些参数是不一样的，会有所改进。正因为如此，所以才会报错，某些代码的函数找不到，或者张量的维度对不上等报错。