YOLOX——Windows10+Anaconda3搭建环境_python

一、Windows下使用Anaconda建立环境

1.创建虚拟环境

2.下载Pytorch

3.配置YOLOX环境

（1）安装YOLOX

（2）安装apex文件

（3）安装pycocotools

（4）下载预训练模型

4.测试

一、Windows下使用Anaconda建立环境

注：电脑已安装Anaconda

1.创建虚拟环境

（1）打开Anaconda Prompt：

（2）创建yolo_x虚拟环境，在上面的窗口中输入：

1）conda create -n yolo_x python=3.7 （创建虚拟环境）

2）activate yolo_x （激活虚拟环境）

注：若已经创建了该虚拟环境，可以使用命令：conda info --envs查看已有虚拟环境

看到yolo_x已存在，可以进行下一步切换到虚拟环境yolo_x：

3）conda activate yolo_x

2.下载Pytorch

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

测试是否安装成功：

（1）python

（2）import torch

（3）print(torch.cuda.is_available())

3.配置YOLOX环境（1）安装YOLOX

YOLOX下载地址：https://gitcode.net/mirrors/Megvii-BaseDetection/YOLOX?utm_source=csdn_github_acceleratork

1）可使用如下命令下载YOLOX：

1.git clone git@github.com:Megvii-BaseDetection/YOLOX.git

git clone git@github.com:Megvii-BaseDetection/YOLOX.git报错！

分析原因：git方式下载github代码至本地时需要依赖ssh key，遇到权限不足问题时一般都是SSH key失效或者SSH key不存在，重新创建SSH key一般就可以解决问题。

解决方案：

（1）（前提是已安装好Git，若没安装可参考博文：Windows安装git图文教程_追到乌云的尽头找太阳的博客-CSDN博客_windows安装git）

（2）安装好Git后，参考博文：git clone 解决Permission Denied (publickey)问题_scanklm的博客-CSDN博客_git publickey

创建SSH key

再次git clone后成功！继续：

2.cd YOLOX # 进入YOLOX目录
3.pip install -r requirements.txt # 安装依赖包
4.python setup.py develop # 安装yolox

显示上图表示成功！

（2）安装apex文件

在YOLOX目录下执行如下命令：

1.git clone https://github.com/NVIDIA/apex

又报错了！

在网上找了多个解决方法，最后将https改为http后成功下载了，如下图所示：

安装成功后，YOLOX文件夹下出现apex文件：

然后，继续输入如下命令，完成apex的安装：

2.cd apex
3.python setup.py install

（3）安装pycocotools

pip install pycocotools

（4）下载预训练模型

模型地址：mirrors / Megvii-BaseDetection / YOLOX · GitCode

下载这两个，并将下载得到的预训练模型放在YOLOX文件下：

4.测试

运行测试demo.py

python tools/demo.py image -f exps/default/yolox_s.py -c yolox_s.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu

参数说明
参数	说明
-c	权重的路径
-path	测试图片的路径
-conf	置信度阈值
-nms	nms的IOU阈值
-tsize	测试图片resize的大小
-save_result	是否保存推理结果
--device	选用gpu或cpu推理