Detectron2安装

Detectron2安装

• detectron2 是Facebook开源的CV库，源码和使用方法访问git链接，本文只总结安装方法和Colab的Demo演示。

• 环境要求(Requirements)：

  • python >= 3.6

  • PyTorch 1.3，torchvision 能跟torch匹配 (其他版本没有试过，1.3和1.1版本有一些函数变化，没有实际用过)

  • Opencv

  • pycocotools

    pip install cython; pip install git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
    

  • GCC >= 4.9

• 安装：

  • Linux：官方给出的安装方案，先装完Requirements，再clone安装。

    git clone [email protected]:facebookresearch/detectron2.git
    cd detectron2
    python setup.py build develop
    # or if you are on macOS
    # MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py build develop
    
    # or, as an alternative to `setup.py`, do
    # pip install .
    

    • 由于新环境以及VPN配置，所以全程用 whl 进行的安装，whl在pypi官方下载，这里打包了一份需要的文件，部分几KB的文件没有，有需求的看看

    • 首先安装Anaconda，官网直接下载然后bash anaconda.sh，这个网上教程很多

    • OpenCV可以直接在网站下载whl

    • Requirements可以直接pip安装

    • 最后cd到detectron2的下载目录使用pip安装

    cd detectron2
    pip install -e detectron2
    # 等待 run setup.py develop完成就行
    

    • 到此就安装完成

  • Windows：由于一开始不想用Linux，所以在win10上装的：

     win10环境下安装detectron2会有很多错误，有cuda安装会报错，没有cuda能成功安装，但是运行Demo的时
     候报错没有cuda，主要还是C++的问题，官方并没有说明，只说可以参考mask-r-cnn benchmark，因为有人在
     win10下安装成功了mask-r-cnn，但是实际并不好改。
     有需要参考链接https://github.com/facebookresearch/detectron2/issues/9
    

    • 首先Requirements，PyTorch和opencv可以直接pip，或者在网站下载opencv和PyTorch。

    • pycocotools：官方cocoapi win下安装不上，不知道为什么，最后参考了mask-r-cnn benchmark的issue和CSDN的解答，安装win版本

      pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
      

    • 安装时候提醒C++版本的问题，装个VS就行，一般装2017版本就够了，有issue提到cuda10.1版本要2019，没有实测。

    • 最后就是下载detectron2，打开git链接，download下来解压，或者git clone，cd到下载目录，在cmd运行

      pip install -e detectron2