setup_faster_rcnn.sh

set -e

if [ ! -d "$HOME/code/py-faster-rcnn" ]; then
    cd $HOME/code
    git clone --recursive git@github.com:amsword/py-faster-rcnn.git
fi

cd ~/code/py-faster-rcnn/caffe-fast-rcnn
if [ ! -f "Makefile.config" ]; then
    cp Makefile.config.example Makefile.config
    echo "WITH_PYTHON_LAYER := 1" >> Makefile.config
    echo "USE_CUDNN := 1" >> Makefile.config

    # for ubuntu 16.04
    echo "INCLUDE_DIRS += /usr/include/hdf5/serial" >> Makefile.config
    echo "LIBRARY_DIRS += /usr/lib/x86_64-linux-gnu/hdf5/serial" >> Makefile.config
fi

make -j
make pycaffe

cd ~/code/py-faster-rcnn/lib
make

cd ~/code/py-faster-rcnn

./data/scripts/fetch_faster_rcnn_models.sh

mkdir -p ~/data
cd ~/data
if [ ! -f "VOCtrainval_06-Nov-2007.tar" ]; then
    wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
    wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
    wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
    tar xvf VOCtrainval_06-Nov-2007.tar
    tar xvf VOCtest_06-Nov-2007.tar
    tar xvf VOCdevkit_08-Jun-2007.tar
fi

cd ~/code/py-faster-rcnn/data
ln -s ~/data/VOCdevkit VOCdevkit2007