win10下 detectron2的安装和测试

1、环境依赖：

cuda 10.1

cudnn 7.6.5

python>=3.6

pytorch>=1.6 (实测最好使用1.6   1.7会报很多错，安装失败)

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

vs 2017   (大部分推荐的都是2019，本人是用的2017成功了)

安装COCOAPI（WINDOWS）

下载以下链接的cocoapi，因为cocoapi官方暂不支持Windows
https://github.com/philferriere/cocoapi 点击跳转.
然后用以下指令安装
cd coco/PythonAPI
python setup.py build_ext --inplace
python setup.py build_ext install

如果报cl: 命令行 error D8021 :无效的数值参数“/Wno-cpp” 和 cl: 命令行 error D8021 :无效的数值参数“/Wno-unused-function”，则打开setup.py，删除如下参数

打开cmd输入
import pycocotools;
from pycocotools.coco import COCO
没报错即成功

安装FVCORE：

下载链接：https://github.com/facebookresearch/fvcore 点击跳转.
然后cd到setup.py所在目录，用以下指令安装
python setup.py build --force develop
打开cmd输入 import fvcore 没报错即成功

 

 

2、安装detectron2

下载源码： https://github.com/facebookresearch/detectron2 

进入到detectron2-master目录下，在地址栏输入cmd回车

 如果pytorch安装在虚拟环境中，先切换到虚拟环境，然后输入  python setup.py build develop

安装成功后 可以输入 pip list查看

3、测试

进入到detectron2-master目录下，在地址栏输入cmd回车

 如果pytorch安装在虚拟环境中，先切换到虚拟环境

python demo/demo.py  --config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml  --webcam  --confidence-threshold 0.5  --opts MODEL.WEIGHTS demo/model_final_280758.pkl

测试报错：

ModuleNotFoundError: No module named 'win32con'

解决办法：
conda install scrapy

使用detectron2训练自己数据集 请移步  https://blog.csdn.net/d597797974/article/details/117091149

到此完结！