YOLOv5 + Win10 训练DJI官方机器人数据集【1】YOLOv5环境搭建

目录

一、环境搭建

1.1安装列表

1.2 安装

1.2.1 Anaconda安装

1.2.2 CUDA10.2 安装

1.2.3 Pytorch安装

1.2.4 Opencv安装

1.2.5 YOLOv5 安装

2、测试yolov5

二、可能遇见的问题

---

初试完，回到实验室再次接触到本科做过的RM比赛，想玩一玩yolov5，搞了一晚上总算能训练了（虽然原理还不太明白qwq），中间遇到了很多坑。由于懒得用ubuntu了，就直接搞了win10版的。下面就是我配置的过程，给大家排排坑。

一、环境搭建

1.1安装列表

（1）N卡

（2）python 3.8以上（anaconda）

（3）cuda 10.2

（4）pytorch

（5）YOLOv5源码

1.2 安装

废话不多说，直接安装。

1.2.1 Anaconda安装

主要是利用anaconda来搭建python、pytorch等环境。

anaconda官方链接https://www.anaconda.com/products/individual，64位选第一个，32位选第二个。

安装成功后，打开这个：

选择这个进入终端：

或者直接打开这个进入终端：

然后执行下面两条指令，安装git和pip：

conda install git

conda install pip

1.2.2 CUDA10.2 安装

不知道现在pytorch支不支持CUDA11，但是这里安装CUDA10.2是能用的。

CUDA10.2官方链接：https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

选择win10的下载安装就可以了。

1.2.3 Pytorch安装

官方链接：https://pytorch.org/get-started/locally/

选择如图所示选项就可以了，然后在刚刚打开的CMD.exe Prompt终端，执行Run this Command:的指令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

1.2.4 Opencv安装

打开anaconda navigator，点选下图所示Enviroments，然后点击Update index更新可下载库文件的清单。选择libopencv、opencv、py-opencv。（不用管我多余的那一个）

1.2.5 YOLOv5 安装

下载YOLOv5前先给YOLOv5项目单独创建环境：

打开anaconda navigator选择CMD.exe Prompt

依次执行以下指令：

conda create -n yolov5 python=3.8  //创建名为【yolov5】的python3.8环境，并通过

conda activate yolov5                        //通过activate激活环境

git clone https://github.com/ultralytics/yolov5.git   //下载yolo v5项目

cd yolov5                                             //将当前路径切换至 yolo v5目录

pip install -U -r requirements.txt       //根据作者创建好的txt，下载所需环境

这里可能会遇到下载速度慢的问题，具体解决方案请看后面的Q1。

2、测试yolov5

打开CMD.exe Prompt，依次执行：

activate yolov5

cd yolov5

python detect.py --source 0

这里执行最后一行代码可能会自动下载官方的模型.pt，然后运行模型。有的可能会下载失败，运行失败，但不代表yolov5安装失败，只要下载完整的.pt就可以了，当然也可以不弄。

以上就是安装yolov5完整过程了，关于数据的训练下一篇更新。

 

二、可能遇见的问题

Q1：这里可能会遇到下载慢的问题，尤其执行 pip install -U -r requirements.txt 要下很多依赖包，速度太太太太太太太太慢了，所以可以挂个外网（不推荐，没啥用）或者换个pip源。

A：在C盘用户里创建文件夹pip

创建pip.ini文件

文件内写入：

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

[install]
trusted-host=mirrors.aliyun.com

之后再执行 pip install -U -r requirements.txt 你就会有种单车变摩托的感觉。

以上就是YOLOv5安装完整过程，数据集训练在下一篇更新。

---

参考博客：https://blog.csdn.net/kasaiki/article/details/108653310

感谢以上大佬分享！