INSTALL.md
September 27, 2019 ยท View on GitHub
- Clone the repo:
git clone https://github.com/ZENGXH/DMM_Net.git
Install PyTorch with CUDA
conda create --name dmm --file requirements.txt -c pytorch -c conda-forge
pip install cython
pip install pycocotools pyyaml yacs opencv-python scikit-image easydict prettytable lmdb tabulate tqdm munkres tensorboardX
# dont install torchvision with pip, otherwise it may download torch 1.2.0 auto
- options:
pip install ninja yacs cython matplotlib easydict prettytable tabulate tqdm ipython scipy opencv-python networkx scikit-image tensorboardx cython scipy pillow h5py lmdb PyYAML
conda install -c pytorch pytorch-nightly cudatoolkit=9.0 -y
Install maskrcnn-benchmark
cd ..
git clone https://github.com/ZENGXH/maskrcnn-benchmark.git
cd maskrcnn-benchmark
## install pkgs
# git clone https://github.com/pytorch/vision.git
# cd vision
# python setup.py install
# cd ..
#
# git clone https://github.com/cocodataset/cocoapi.git
# cd cocoapi/PythonAPI
# python setup.py build_ext install
# cd ../../
# build maskrcnn-benchmark
python setup.py build develop