Update install process

Author: ericsujw

README.md

@@ -19,6 +19,7 @@ Image colorization is inherently an ill-posed problem with multi-modal uncertain
 In IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2020.
 
 ## Prerequisites
+* [CUDA 10.1](https://developer.nvidia.com/cuda-10.1-download-archive-update2)
 * Python3
 * Pytorch >= 1.5
 * Detectron2
@@ -35,12 +36,16 @@ cd InstColorization
 2. Install [conda](https://www.anaconda.com/).
 3. Install all the dependencies
 ```sh
-conda env create --file env.yml --name instacolorization
+conda env create --file env.yml
 ```
 4. Switch to the conda environment
 ```sh
 conda activate instacolorization
 ```
+5. Install other dependencies
+```sh
+sh scripts/install.sh
+```
 
 ## Pretrained Model
 1. Download it from [google drive](https://drive.google.com/open?id=1Xb-DKAA9ibCVLqm8teKd1MWk6imjwTBh).

env.yml

@@ -1,4 +1,4 @@
-name: py37_pt14
+name: instacolorization
 channels:
   - conda-forge
   - defaults
@@ -92,8 +92,6 @@ dependencies:
     - absl-py==0.9.0
     - cachetools==4.1.0
     - chardet==3.0.4
-    - detectron2==0.1.2+cu101
-    - dominate==2.4.0
     - future==0.18.2
     - fvcore==0.1.dev200506
     - google-auth==1.14.2
@@ -111,9 +109,7 @@ dependencies:
     - protobuf==3.11.3
     - pyasn1==0.4.8
     - pyasn1-modules==0.2.8
-    - pycocotools==2.0
     - pydot==1.4.1
-    - pyyaml==5.1
     - pyzmq==18.1.1
     - requests==2.23.0
     - requests-oauthlib==1.3.0
@@ -122,9 +118,6 @@ dependencies:
     - tensorboard==2.2.1
     - tensorboard-plugin-wit==1.6.0.post3
     - termcolor==1.1.0
-    - torch==1.5.0+cu101
-    - torchfile==0.1.0
-    - torchvision==0.6.0+cu101
     - urllib3==1.25.8
     - visdom==0.1.8.9
     - websocket-client==0.57.0

scripts/install.sh

@@ -0,0 +1,5 @@
+pip install -U torch==1.5 torchvision==0.6 -f https://download.pytorch.org/whl/cu101/torch_stable.html
+pip install cython pyyaml==5.1
+pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
+pip install dominate==2.4.0
+pip install detectron2==0.1.2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/index.html