win10下detectron2环境配置

代码下载自2020-06-17更新代码。若代码结构不一致变化较大时,可对代码回滚至当前版本:

git reset --hard ae931b4

1、系统环境

win10、vs2017、cuda10.1、pytorch1.4、torchvision0.5。

2、配置conda虚拟环境

conda create --n detectron2 python=3.6
conda activate detectron2
conda install pytorch=1.4 torchvision=0.5 cudatoolkit=10.1
conda install pytorch-nightly  ipython

# dependencies
pip install ninja yacs cython matplotlib==2.2.4 tqdm opencv-python

python setup.py build develop

编译后,缺少包,进行安装:

pip install fvcore tensorboard future pydot mock
conda install pillow # PIL包,windows上需要编译,通过conda安装

重新编译,通过:

python setup.py build develop

注:6月4号后,版本若cuda编译失败,那是因为AMD公司提交的一条记录,修改cuda编译引起的。代码回滚值前一条记录,替换对应csrc文件即可通过。

编译成功后:

detectron2配置安装,总目录:

1、coco数据集下载格式解析

2、win10下detectron2环境配置

3、windows10离线安装pycocotools

4、detectron2代码运行及数据集配置

5、detectron2使用自定义数据集及数据加载

6、detectron2模型使用、读写、训练及测试

7、detectron2配置与部署

 

 

你可能感兴趣的:(detectron2)