-
Notifications
You must be signed in to change notification settings - Fork 6
Extract module installation
Ostrzyciel edited this page Feb 12, 2020
·
1 revision
These instructions were not tested, we found them lying around in some old docs. They may not work at all, but that's all we have.
- Pytorch 1.0, build nightly
- torchvision
- cocoapi
- yacs
- matplotlib
- GCC >= 4.99
- Create a conda environment
conda create --name maskrcnn_benchmark
conda activate maskrcnn_benchmark
- Install python
conda install ipython
- Install needed libraries
pip install ninja yacs cython matplotlib tqdm numpy opencv-python
- Install PyTorch, below an example for CUDA 9.0. Instructions for other CUDA versions can be found here: https://pytorch.org/get-started/locally/
conda install -c pytorch pytorch-nightly torchvision cudatoolkit=9.0
export INSTALL_DIR=$PWD
- Install pycocotools
cd $INSTALL_DIR
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install
- Download the code
cd $INSTALL_DIR
git clone https://github.com/mini-pw/2019L-ProjektZespolowy/tree/master/
cd 2019L-ProjektZespolowy/extract/model/maskrcnn-benchmark
- Install the model
python setup.py build develop
unset INSTALL_DIR