detectron2 安装

一、detectron2

下载:地址

 

环境:

  • anaconda3
  • cuda10.1 (10.0应该也可以)
  • python3.6
  • torch >= 1.3
  • opencv 4.1

 

二、win10 版本 安装

1)环境

  • vs2019

 

2) 确保 cl 有效

C:\Users\yuki>cl
用于 x64 的 Microsoft (R) C/C++ 优化编译器 19.00.24245 版
版权所有(C) Microsoft Corporation。保留所有权利。

用法: cl [ 选项... ] 文件名... [ /link 链接选项... ]

若不能调用 环境变量添加:

添加内容(按实际版本):C:\Program Files (x86)\Microsoft Visual Studio\2019\Professional\VC\Tools\MSVC\14.23.28105\bin\Hostx86\x64

 

2)  安装 cocoapi 社区版

  • 下载:地址
  • 安装:(建议保持以管理员模式运行cmd)
cd PythonAPI
python setup.py build_ext --inplace
python setup.py build_ext install

 

3) 安装 fvcore

  • 下载:地址
  • 安装:(建议保持以管理员模式运行cmd)
python setup.py build --force develop

 

4)安装 ninja

相关文章:地址

 

5)安装 detectron2 (管理员 cmd 打开窗口)

cd detectron2

python -m pip install -e .   #下载速度满 补充  -i https://mirrors.aliyun.com/pypi/simple

 

三、unbuntu 版本 安装 ()

1)  安装 ninja

相关文章:地址

 

2)安装 detectron2 

# v0.3 适用
python -m pip install git+https://github.com/facebookresearch/detectron2.git #已下载可忽略

cd detectron2

python -m pip install -e .   #下载速度满 补充  -i https://mirrors.aliyun.com/pypi/simple

 

四、已测试系统

  • win10
  • ubuntu18.04
  • Jetpack4.4

 

五、测试

模型库:地址

python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml
–input input1.jpg input2.jpg
–opts MODEL.WEIGHTS mask_rcnn_R_50_FPN.pkl # 模型地址

 

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页