diff --git a/INSTALL.md b/INSTALL.md index ef07481..31102ea 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -2,16 +2,16 @@ ``` # install pytorch 1.1 and torchvision -sudo pip3 install torch==1.1 torchvision +sudo pip3 install torch==1.1 torchvision==0.3.0 # install apex cd $INSTALL_DIR git clone https://github.com/NVIDIA/apex.git cd apex -sudo python setup.py install --cuda_ext --cpp_ext +pip3 install -v --no-cache-dir ./ # clone Hier-R-CNN -git clone https://github.com/soeaver/Parsing-R-CNN.git +git clone https://github.com/soeaver/Hier-R-CNN.git # install other requirements pip3 install -r requirements.txt @@ -21,10 +21,10 @@ cd Hier-R-CNN sh make.sh # make cocoapi -cd Parsing-R-CNN/cocoapi/PythonAPI -mask -cd ../../ -ln -s cocoapi/PythonAPI/pycocotools/ ./ +cd Hier-R-CNN/cocoapi/PythonAPI +make +cd ../../../ +ln -s Hier-R-CNN/cocoapi/PythonAPI/pycocotools/ ./ ``` ## Data and Pre-train weights diff --git a/README.md b/README.md index ba7c5ff..b582fbb 100644 --- a/README.md +++ b/README.md @@ -45,7 +45,7 @@ You need to download the datasets and annotations following this repo's formate. - [CIHP](https://drive.google.com/open?id=1OLBd23ufm6CU8CZmLEYMdF-x2b8mRgxV) -- [MHP-v2](coming soon) +- [MHP-v2](https://lv-mhp.github.io/dataset) - [DensePoseData](https://drive.google.com/open?id=1WiTLYVIgMyCDENXHPVEWW7qbZ-3EBjbt)(using original [MSCOCO2017](http://cocodataset.org/#download) images)