Skip to content

Instantly share code, notes, and snippets.

@michhar
Last active June 10, 2021 06:32
Show Gist options
  • Save michhar/a4086aec42eb9d020d47288d0987a92e to your computer and use it in GitHub Desktop.
Save michhar/a4086aec42eb9d020d47288d0987a92e to your computer and use it in GitHub Desktop.
Notes on installing pytorch-mask-rcnn

Following https://github.com/multimodallearning/pytorch-mask-rcnn#installation

Setup

Caveats

  • Using Anaconda3 instead of Miniconda3
  • For nms build: nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -ccbin "c:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin -arch=sm_61
  • updated path to include c:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin
  • Set the following environmental vars (can set in the System Variables as well):
    • set INCLUDE=\Program Files (x86)\Windows Kits\10\Include\10.0.10240.0\ucrt according to this SO post
    • set CMAKE_GENERATOR=Visual Studio 15 2017 Win64
    • set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build"
    • set DISTUTILS_USE_SDK=1
    • set LIB=\Program Files (x86)\Windows Kits\10\Lib\10.0.10240.0\um\x86;\Program Files (x86)\Windows Kits\10\Lib\10.0.10240.0\ucrt\x86
    • set PATH=C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin;%PATH%
  • call "C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build\vcvarsall.bat" x86 -vcvars_ver=14.12

Coco API build

  • git clone: https://github.com/cocodataset/cocoapi and python setup.py install in PythonAPI library
  • In Git Bash linked pycocotools with: ln -s cocoapi/PythonAPI/pycocotools/ pycocotools
  • To get this PythonAPI to compile:
First go to cocoapi\PythonAPI\setup.py and change line 14 from:
extra_compile_args=['-Wno-cpp', '-Wno-unused-function', '-std=c99'],
to
extra_compile_args={'gcc': ['/Qstd=c99']},
as pointed out in this issue: CharlesShang/FastMaskRCNN#173

From: cocodataset/cocoapi#51

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment