diff --git a/INSTALL.md b/INSTALL.md index b1bfaa293..2544ff965 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -16,48 +16,34 @@ # first, make sure that your conda is setup properly with the right environment # for that, check that `which conda`, `which pip` and `which python` points to the # right path. From a clean conda env, this is what you need to do - conda create --name maskrcnn_benchmark conda activate maskrcnn_benchmark # this installs the right pip and dependencies for the fresh python conda install ipython -# maskrcnn_benchmark and coco api dependencies -pip install ninja yacs cython matplotlib tqdm opencv-python - # follow PyTorch installation in https://pytorch.org/get-started/locally/ # we give the instructions for CUDA 9.0 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 # install apex -cd $INSTALL_DIR git clone https://github.com/NVIDIA/apex.git cd apex python setup.py install --cuda_ext --cpp_ext # install PyTorch Detection -cd $INSTALL_DIR -git clone https://github.com/facebookresearch/maskrcnn-benchmark.git -cd maskrcnn-benchmark - -# the following will install the lib with -# symbolic links, so that you can modify -# the files if you want and won't need to -# re-build it -python setup.py build develop - - -unset INSTALL_DIR +pip install git+https://github.com/facebookresearch/maskrcnn-benchmark +# for development on mask-rcnn benchmark, clone the repository and install in development mode: +# the files if you want and won't need to re-build it: +# git clone https://github.com/facebookresearch/maskrcnn-benchmark.git +# cd maskrcnn-benchmark.git +# 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 ``` @@ -71,9 +57,6 @@ conda activate maskrcnn_benchmark # this installs the right pip and dependencies for the fresh python conda install ipython -# maskrcnn_benchmark and coco api dependencies -pip install ninja yacs cython matplotlib tqdm opencv-python - # follow PyTorch installation in https://pytorch.org/get-started/locally/ # we give the instructions for CUDA 9.0 ## Important : check the cuda version installed on your computer by running the command in the cmd : diff --git a/setup.py b/setup.py index 837c2cd15..f66ad4194 100644 --- a/setup.py +++ b/setup.py @@ -11,7 +11,7 @@ from torch.utils.cpp_extension import CppExtension from torch.utils.cpp_extension import CUDAExtension -requirements = ["torch", "torchvision"] +requirements = ["torch", "torchvision", "ninja", "yacs", "cython", "matplotlib", "tqdm", "opencv-python"] def get_extensions(): @@ -63,7 +63,7 @@ def get_extensions(): url="https://github.com/facebookresearch/maskrcnn-benchmark", description="object detection in pytorch", packages=find_packages(exclude=("configs", "tests",)), - # install_requires=requirements, + install_requires=requirements, ext_modules=get_extensions(), cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension}, )