YOLOv5环境配置

0 安装Anaconda和pycharm

略…

1 创建YOLOv5虚拟环境

  1. 找到安装好的Anaconda,打开Anaconda Prompt

YOLOv5环境配置_第1张图片

  1. 在命令面板中输入一下代码
conda create -n yolov5 python=3.8
  1. 输入y

YOLOv5环境配置_第2张图片

  1. 出现下图表示创建完成

YOLOv5环境配置_第3张图片

  1. 输入activate yolov5激活环境(该窗口不要关闭)

YOLOv5环境配置_第4张图片

2 安装pytorch

2.1 查看电脑支持CUDA版本

  1. 电脑桌面右键,打开NVIDIA控制面板

YOLOv5环境配置_第5张图片

  1. 在【帮助】菜单拦下,打开【系统信息】

YOLOv5环境配置_第6张图片

  1. 在【组件】窗口,查看自己的CUDA支持版本

我这里最高支持11.4版本

YOLOv5环境配置_第7张图片

2.2 下载pytorch

  1. 进入PyTorch官网,选择适合自己的版本

需要说明一下,30系列显卡,必须安装11以上的版本,10版本不再支持

YOLOv5环境配置_第8张图片

  1. 这里我选择安装11.3版本

YOLOv5环境配置_第9张图片

  1. 在开始之前,先将pip下载切换为国内镜像(在 pip install 后面添加 -i https://pypi.tuna.tsinghua.edu.cn/simple)

这里选择清华源,临时镜像,如果需要换成永久的,可以进清华源的官网根据提示进行切换

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

image-20230803113738501

  1. 内容有点大,耐心等待安装完成

出现successfully表示安装成功

YOLOv5环境配置_第10张图片

3 下载yolov5源码

GitHub地址:YOLOv5

国内GitHub访问慢,可以使用Watt Toolkit软件进行加速(这个不是梯子,免费的,加速功能强大)

  1. 进入GitHub后,在右侧找到releases,进入下载zip(下载的v7)

YOLOv5环境配置_第11张图片

YOLOv5环境配置_第12张图片

  1. 找一个文件夹,将下载的压缩包解压出来

YOLOv5环境配置_第13张图片

4 PyCharm部署运行

  1. 用PyCharm打开刚才解压出来的文件夹

YOLOv5环境配置_第14张图片

  1. 先不创建虚拟环境,进入后手动添加环境

YOLOv5环境配置_第15张图片

YOLOv5环境配置_第16张图片

YOLOv5环境配置_第17张图片

YOLOv5环境配置_第18张图片

  1. 打开requirements.txt文件,将下图所示的部分进行修改

YOLOv5环境配置_第19张图片

  1. 在 PyCharm打开终端,输入activate yolov5激活环境

YOLOv5环境配置_第20张图片

YOLOv5环境配置_第21张图片

  1. 安装依赖环境
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  -r requirements.txt 

安装过程中,会出现一堆warring,这个不用管

出现下图所示内容代表安装成功

YOLOv5环境配置_第22张图片

  1. 运行测试
python detect.py

到下图这一步,下载较慢,这时可以将下载地址复制下来,在浏览器上下载

image-20230803123957619

输入Ctrl+C,结束命令行下载任务

image-20230803124314690

将刚才下载的文件放到文件夹下,然后在pycharm命令行中重新执行python detect.py

YOLOv5环境配置_第23张图片

出现下图信息,表示环境搭建成功

YOLOv5环境配置_第24张图片

打开runs文件夹夹,监测效果如图所示

python detect.py

YOLOv5环境配置_第25张图片

出现下图信息,表示环境搭建成功

YOLOv5环境配置_第26张图片

打开runs文件夹夹,监测效果如图所示

YOLOv5环境配置_第27张图片

你可能感兴趣的:(笔记,人工智能,YOLO,深度学习,python)