【Windows配置一】Python简单运行YOLOv5

一. PyTorch环境配置及安装

1.软件安装

安装Anaconda3参考：https://blog.csdn.net/qq_36015370/article/details/79484455
其他部分参考： https://blog.csdn.net/zhouzhiyao960211/article/details/102873717.

注意：3060显卡选择pytorch的v1.80版本的CUDA 11.1

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

图为Anaconda3默认安装位置

其他快速安装pytorch方法
参考：http://t.csdn.cn/NhfIt

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.6.1 -f https://download.pytorch.org/whl/torch_stable.html

2. Pytorch环境安装

1、执行如下的指令查看有哪些环境

conda env list

结果：

2、创建虚拟环境conda create -n 环境名字（英文） python=x.x（python版本），如下，我就是创建了一个名字叫YOLOv5 ，python是3.8版本的环境。

conda create -n YOLOv5 python=3.8

结果：

3. pip清华镜像源配置

参考：http://t.csdn.cn/UfkCp

1、临时使用

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

把some-package替换为所需模块即可（还真有个模块叫some-package）

2、永久配置

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

如果报错，可能是因为pip版本不够（需≥10.0.0），先升级pip：

pip install pip -U

当然，升级pip本身也可以使用清华镜像源。

二. 安装YOLOv5依赖

1、按照 “requirements.txt” 安装依赖

//第一步
cd "requirements.txt"文件路径 
#例如：cd C:\Users\LZY\Desktop\yolov5\yolov5
//第二步
conda activate pytorch
//第三步
pip install -r requirements.txt
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/  -r requirements.txt

2、安装结果

安装结果给出了pycharm的python配置环境的路径，与上面的环境配置路径一致
上图中的 WARNING: Ignoring invalid distribution...问题，参考：https://blog.csdn.net/Scarlett2025/article/details/117995427.
警告原因：之前安装插件失败/中途退出，导致插件安装出现异常导致的
解决方法：找到警告信息中报错的目录，然后删掉 “~开头” 的文件夹

三. pycharm打开YOLOv5

1、打开 “文件—设置—Python解释器” 界面，配置python

2、将"当前文件夹"设置为"源根"

链接: https://blog.csdn.net/qq_30622831/article/details/80978118
这一步是解决：pycharm导入本地py文件时，模块下方出现红色波浪线

四. 简单测试

注意：运行v5报以下错误，参考链接:
https://blog.csdn.net/RooKichenn/article/details/120866650.

如果你用的是v5的话,就去v6里面的models/common.py里面去找到这个SPPF的类,把它拷过来到你这个Tags5的models/common.py里面,这样你的代码就也有这个类了,还要在Tags5的models/common.py里引入一个warnings包就可以了

五、安装labelimg

1、注意：首先将pip升级一下，否则有可能出现一些奇奇怪怪的问题
比如报下面错误

Could not find a version that satisfies the requirement sip<7,>=6.4

2、这里主要讲的是在window系统中的安装，首先打开cmd命令行（快捷键：win+R）。进入cmd命令行控制台。输入如下的命令：

pip install labelimg -i https://pypi.tuna.tsinghua.edu.cn/simple

3、有可能出现的问题
（1）按照上述安装有可能会报错：

解决办法：
参考：链接: https://blog.csdn.net/lsf_007/article/details/87931823.
这里利用阿里云安装

python -m pip install --user labelimg -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

（2）安装labelimg过程中出现ModuleNotFoundError: No module named ‘PyQt4’
参考：https://zhuanlan.zhihu.com/p/377302284

// 卸载PyQt5
pip uninstall PyQt5
// 安装python-qt5以获取PyQt5
pip install --user python-qt5