0 前言
由于YOLOx是基于Linux系统编写的代码,如果要跑通demo.py,需要对代码进行修改。
在论坛上看了一圈,记录Windows系统的很少,所以有了这篇文章。
本文为Windows部署YOLOx的踩坑记录,本文将详细介绍跑通demo的过程和个人遇到的所有报错。
1 准备工作
1.1 必要环境
Windows+Anaconda+PyCharm+Python
1.1.1 Anaconda创建新环境(Python3.8)
在Anaconda Prompt终端输入如下代码创建新的虚拟环境。
conda create -n yolo_x python=3.8
输入y
等待下载,创建新环境,可能要等个几分钟。
出现上图创建成功。
终端输入conda activate yolo_x
激活新环境。
成功进入新环境。
1.1.2 安装PyTorch
接下来打开PyTorch官网,官网地址:PyTorch官网地址
这里直接通过pip指令安装PyTorch。博主的CUDA是10.2的,所以安装10.2版本,可以根据自己的需求来。
复制Run this Command中的指令到终端。
pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
等待一段时间安装。
安装过后,可以看到一片红。
1.1.3 其他库的安装
不要慌张,不是安装失败了。既然提示我们缺失库,那就根据缺失的部分自行进行安装,
在终端输入
pip install cloudpickle
pip install h5py
pip install imageio
pip install requests
pip install scikit-image
pip install scikit-learn
pip install wrapt
安装完后,可以看到已经没有错误了。
这样新环境就创建成功了,接下来,我们可以继续安装一些比较重要的库,比如
openCV:pip install opencv-python
Pandas:pip install pandas
Numpy:pip install numpy
MatplotLib:pip install matplotlib
后面这两个库其实已经预装好了,可以不用再自己安装。
在终端输入pip list
,可以看到自己在新虚拟环境中所有安装的库列表。
这样我们的新环境就已经搭建好啦。
1.2 配置YOLOx环境
1.2.1 下载YOLOx源码
需要到Github官网上下载YOLOx源码,官网地址:YOLOx下载地址
点击Code
>Download ZIP
下载压缩包。
上不去的同学去找个梯子哦。如果有需要的话可以留言,我之后放个百度云链接上来。
1.2.2 PyCharm平台配置虚拟环境
下载好了之后,解压YOLOX-main.zip
这个文件。
打开PyCharm,点击打开
>找到解压后的文件夹,将整个项目添加到平台上。
加下来配置虚拟环境,点击文件
>设置
>项目:YOLOx-main
>python解释器
>python解释器后面的设置按钮
点击设置按钮后,添加
>Conda环境
>现有环境
>确定
>应用
>确定
一定要确定解释器是自己创建环境yolo_x
下的python.exe
接下来就是等待PyCharm更新Python解释器。
1.2.3 安装依赖requirements.txt
可以看到有一个requirements.txt
文件,这里面装有跑通程序所需要的库,我们需要把这些库都装上。
打开终端输入activate yolo_x
进入环境。
可以看到现在在C盘下,而文件是在D盘下,那就需要进入D盘,在终端输入d:
指令进入D盘。
输入指令cd Download\YOLOX-main\YOLOX-main
,这里download换成你们自己的文件夹。
进入后输入指令pip install -r requirements.txt
。
这样requirements.txt依赖就安装完成啦,这一步一般是不会出错,有出错的可以留言。
1.2.3 根据官网步骤安装环境
-
安装YOLOx:终端输入指令
python setup.py install
(注意:安装YOLOx仍然是要在Download\YOLOX-main\YOLOX-main
这个文件夹下进行,因为setup.py
文件在该路径下。) -
安装apex
安装到官网下载apex,解压到项目文件夹内
官网下载地址:apex下载地址
在终端输入指令cd apex-master
进入文件夹
继续输入指令python setup.py install
安装apex