caffe-lrp-cpu-u16.04.def

Bootstrap: docker
From: ubuntu:16.04

# This definition file allows the user to build a lrp toolbox executable for caffe, based on an ubuntu 16.04
# docker image. E.g. when calling
# > singularity build --fakeroot --force caffe-lrp-cpu-u16.04.sif caffe-lrp-cpu-u16.04.def
# the resulting sif file can directly be called from command line, e.g. as
# > ./caffe-lrp-cpu-u16.04.sif -c caffe-lrp-config-file.txt -t input-file-list.txt
# on any linux-based OS.


%post

  ####
  echo ">>> STEP 0: DISTRO PACKAGE INSTALLATION"
  ####

  # install caffe-related dependencies. order of installation must be exactly this. else, it will break down.
  apt-get update
  apt-get install -y libleveldb-dev \
                     libprotobuf-dev \
                     libsnappy-dev \
                     libopencv-dev \
                     libhdf5-serial-dev \
                     protobuf-compiler \
                     build-essential \
                     libatlas-base-dev \
                     libgflags-dev \
                     libgoogle-glog-dev \
                     liblmdb-dev \
                     libmagick++-dev

  apt-get update
  apt-get install -y --no-install-recommends libboost-all-dev

  apt-get update
  apt-get install -y apt-utils \
                     wget \
                     sed \
                     git \
                     tree \
                     unzip


  ####
  echo ">>> STEP 1: DOWNLOAD RELEVANT REPOSITORIES"
  ####
  cd /opt
  BRANCH=master
  wget https://github.com/sebastian-lapuschkin/lrp_toolbox/archive/$BRANCH.zip
  unzip $BRANCH.zip
  mv lrp_toolbox-$BRANCH lrp_toolbox
  rm $BRANCH.zip


  ####
  echo ">>> STEP 2: APPLY DISTRO-SPECIFIC PATCH TO CAFFE"
  ####
  # enter the caffe lrp folder, patch code for ubuntu 14.04->16.04 and install.
  cd /opt/lrp_toolbox/caffe-master-lrp

  # include "hdf5.h" -> #include "hdf5/serial/hdf5.h"
  # include "hdf5_hl.h" -> #include "hdf5/serial/hdf5_hl.h"
  # links hdf5 to where it can be found by caffe make and modifies makefile.
  # adapts include paths and library names for the demonstrator
  echo "    running patch."
  echo "    step 2.1: manipulate header paths for hdf5 "
  find ./*/ -type f -exec sed -i -e 's^"hdf5.h"^"hdf5/serial/hdf5.h"^g' -e 's^"hdf5_hl.h"^"hdf5/serial/hdf5_hl.h"^g' '{}' \;

  echo "    step 2.2: create symlinks for hdf5 libraries. please enter your super user pw on prompt "
  here=$PWD
  cd /usr/lib/x86_64-linux-gnu
  ln -s libhdf5_serial_hl.so libhdf5_hl.so
  ln -s libhdf5_serial.so libhdf5.so
  echo "        going back to $here"
  cd $here

  echo "    step 2.3: modify makefile to include hdf5 dirs"
  sed -i -e "/INCLUDE_DIRS :=/a INCLUDE_DIRS += \/usr\/include\/hdf5\/serial\/" Makefile.config

  echo "    step 2.4: modify imagemagick include paths and library names for demonstrator"
  cd demonstrator
  sed -i -e "s/-lMagick++/-lMagick++-6.Q16/g" -e "s/-lMagickWand/-lMagickWand-6.Q16/g" -e "s/-lMagickCore/-lMagickCore-6.Q16/g" -e "s|-I /usr/include/ImageMagick/|-I /usr/include/ImageMagick-6 -I /usr/include/x86_64-linux-gnu/ImageMagick-6/|g" build.sh
  cd ..


  ####
  echo ">>> STEP 3: BUILD CAFFE FOR LRP"
  ####
  # run makefile and test.
  make clean
  make all -j10
  # make test
  # make runtest # NOTE: tests from LayerFactoryTest/ will fail. Does not matter though for LRP; but will kill the container building


  ####
  echo ">>> STEP 4: BUILD AND TEST DEMONSTRATOR APPLICATION"
  ####

  cd /opt/lrp_toolbox/caffe-master-lrp/demonstrator
  bash build.sh
  chmod +x lrp_demo

  # DOWNLOAD BVLC REFERENCE CAFFE MODEL AND START DEMONSTRATOR
  bash download_model.sh
  ./lrp_demo ./config_sequential.txt ./testfilelist.txt ./
  echo

  echo "output images can be found in $(pwd)/lrp_output:"
  tree $(pwd)/lrp_output
  cd ..
  echo "If you can read this, the container has been built successfully"



%files
  caffe_lrp_commandline.py /opt/caffe_lrp_commandline.py

%runscript
  python /opt/caffe_lrp_commandline.py "$@"