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
        

你可能感兴趣的:(CV)