windows 10编译安装detectron2（0.2.1）

windows 10编译安装detectron2（0.2.1）

基础环境

anaconda3 pytorch1.6 python 3.6 cuda10.2 vs2019

1、下载detectron2（0.2.1）

https://github.com/facebookresearch/detectron2/tree/v0.2.1

2、创建anaconda虚拟环境

conda create -n detectron2 python==3.6

conda activate detectron2

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

测试pytorch正确安装

import torch

torch.__version__
'1.6.0'
torch.cuda.is_available()
True

3、call VS2019的编译环境

call "D:\Program Files (x86)\Microsoft Visual Studio\2019\Professional\VC\Auxiliary\Build\vcvars64.bat"

cd 到detectron2目录中

set DISTUTILS_USE_SDK=1

python setup.py build develop

4、安装其他包

pip install opencv-python tqdm matplotlib fvcore cython cloudpickle mock tensorboard

装完之后测试

测试demo, 到detectron2 MODEL ZOO下载Faster R-CNN X101-FPN模型权重, 假设放在目录xxx下

python demo/demo.py ^
–config-file configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml ^
–input demo/coco_val2017_000000002153.jpg ^
–opts MODEL.WEIGHTS xxx/faster_rcnn_X_101_32x8d_FPN_3x.pkl

参考链接https://blog.youkuaiyun.com/ying86615791/article/details/109157249