win10下基于anaconda的detectron2安装

最新推荐文章于 2024-06-12 10:33:50 发布

simple_whu

最新推荐文章于 2024-06-12 10:33:50 发布

阅读量2k

点赞数 3

分类专栏： python 文章标签： python 深度学习 pytorch

本文链接：https://blog.csdn.net/qq_42679415/article/details/125899218

版权

39 篇文章 2 订阅

订阅专栏

例如，我给创建的虚拟环境起名叫mPytorch，执行下面的命令以创建这个环境：

conda create -n mPytorch python=3.7

创建完成后，使用conda activate语句激活该环境。当前激活了哪个环境，运行conda install或者pip install安装的库就存在于哪个环境。激活mPytorch后，运行官方命令安装pytorch和cudatoolkit。

conda activate mPytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

注意，上一节cudatoolkit安装了哪个版本，就要去英伟达官网下载与其版本对应的C语言cuda开发包，并安装。其下载安装参考该网站。具体下载地址。
下载完成后，双击该安装包，接受默认安装选项，直至安装完成。

若下载了cuda10.x版本，则应该安装VS2019；若下载了cuda11.x版本，则应该安装VS2022。
安装完成后，应该在开始菜单可以找到x64 Native Tools Command Prompt for VS 2019命令行入口：
在这里插入图片描述

虽然官方声称暂不支持windows下编译安装，但是亲测可行。

在GitHub下载detectron2的源码并解压，或者使用git clone命令将仓库复制到本地（如果你安装了Git for windows的话）。

源代码文件夹如图所示：

在这里插入图片描述
根据官方教程，要编译和安装detectron2，只需要在命令行执行一句命令：

python -m pip install -e .\detectron2-main\

其中-e后面是源码文件夹路径，可以是绝对路径或相对路径。
但是，windows下想要成功完成编译，需要注意以下几点：

pip3 install easydict
conda install pyyaml
conda install cffi
conda install cython
conda install scipy
pip install pycocotools

打开你的x64 Native Tools Command Prompt for VS 2019或x64 Native Tools Command Prompt for VS 2022，并输入conda activate mPytorch激活conda环境。
设置临时环境变量

set DISTUTILS_USE_SDK=1

修改源码文件nms_rotated_cuda.cu:

打开nms_rotated_cuda.cu，将下图中框出部分在行首用’//'注释掉即可。这是为了解决错误信息nms_rotated_cuda.cu(68): error: identifier "single_box_iou_rotated" is undefined，也可以参考此博客。
运行python -m pip install -e .\detectron2-main\进行编译和安装。

需要安装opencv。

pip install opencv-python

根据官方示例，下载一个训练好的模型，model_final_280758.pkl，并把它放在detectron2源码的demo文件夹下。
然后在命令行进入demo文件夹，运行命令：

python .\demo.py --config-file ..\configs\COCO-Detection\faster_rcnn_R_50_FPN_3x.yaml --input 你的输入图片名  --opts MODEL.WEIGHTS model_final_280758.pkl

在这里插入图片描述

关注

专栏目录