run.sh

   1 #!/bin/bash
   2
   3 #########################################################################
   4 # This program is free software: you can redistribute it and/or modify  #
   5 # it under the terms of the version 3 of the GNU General Public License #
   6 # as published by the Free Software Foundation.                         #
   7 #                                                                       #
   8 # This program is distributed in the hope that it will be useful, but   #
   9 # WITHOUT ANY WARRANTY; without even the implied warranty of            #
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU      #
  11 # General Public License for more details.                              #
  12 #                                                                       #
  13 # You should have received a copy of the GNU General Public License     #
  14 # along with this program. If not, see <http://www.gnu.org/licenses/>.  #
  15 #                                                                       #
  16 # Written by Francois Fleuret, (C) IDIAP                                #
  17 # Contact <francois.fleuret@idiap.ch> for comments & bug reports        #
  18 #########################################################################
  19
  20 MAIN_URL="http://www.idiap.ch/folded-ctf"
  21
  22 # Compiling
  23
  24 make -j -k
  25
  26 if [[ $? != 0 ]]; then
  27     echo "Compilation failed." >&2
  28     exit 1
  29 fi
  30
  31 echo
  32
  33 ######################################################################
  34 # Generating the pool file
  35
  36 DATA_PATH=./rmk-data
  37 POOL_NAME=${DATA_PATH}/rmk.pool
  38
  39 if [[ -d ${DATA_PATH} ]]; then
  40
  41     if [[ -f ${POOL_NAME} ]]; then
  42
  43         echo "The pool file exists."
  44
  45     else
  46
  47         echo "Can not find the pool file, checking the data integrity."
  48
  49         md5sum -c ${DATA_PATH}/list.md5
  50
  51         if [[ $? != 0 ]]; then
  52             echo "The data set is corrupted. You can download it" >&2
  53             echo "from ${MAIN_URL}" >&2
  54             exit 1
  55         fi
  56
  57         echo "Generating the pool file."
  58
  59         ./list_to_pool ${DATA_PATH}/full.lst ${DATA_PATH} ${POOL_NAME}.wrk
  60
  61         if [[ $? == 0 ]]; then
  62             mv ${POOL_NAME}.wrk ${POOL_NAME}
  63         else
  64             \rm ${POOL_NAME}.wrk 2> /dev/null
  65             echo "Pool generation failed." >&2
  66             exit 1
  67         fi
  68
  69     fi
  70
  71 else
  72
  73     echo "Can not find the RateMyKitten images in ${DATA_PATH}. You can download" >&2
  74     echo "them from ${MAIN_URL}" >&2
  75     exit 1
  76
  77 fi
  78
  79 ######################################################################
  80 # Running the computation per se
  81
  82 RESULT_DIR=./results
  83
  84 if [[ ! -d ${RESULT_DIR} ]]; then
  85     mkdir ${RESULT_DIR}
  86 fi
  87
  88 for SEED in {0..9}; do
  89
  90     for MODE in hb h+b; do
  91
  92         EXPERIMENT_RESULT_DIR="${RESULT_DIR}/${MODE}-${SEED}"
  93
  94         mkdir ${EXPERIMENT_RESULT_DIR} 2> /dev/null
  95
  96         if [[ $? == 0 ]]; then
  97
  98             OPTS="--random-seed=${SEED} --wanted-true-positive-rate=0.75"
  99
 100             if [[ $MODE == "h+b" ]]; then
 101                 OPTS="${OPTS} --force-head-belly-independence=yes"
 102             fi
 103
 104             if [[ $1 == "light" ]]; then
 105                 OPTS="${OPTS} --nb-classifiers-per-level=1 --nb-weak-learners-per-classifier=10"
 106                 OPTS="${OPTS} --proportion-for-train=0.1 --proportion-for-validation=0.1 --proportion-for-test=0.1"
 107             fi
 108
 109             ./folding \
 110                 --niceness=15 \
 111                 --pool-name=${POOL_NAME} \
 112                 --nb-levels=2 \
 113                 --nb-classifiers-per-level=25 --nb-weak-learners-per-classifier=100 \
 114                 --result-path=${EXPERIMENT_RESULT_DIR} \
 115                 --detector-name=${EXPERIMENT_RESULT_DIR}/default.det \
 116                 ${OPTS} \
 117                 open-pool \
 118                 train-detector \
 119                 compute-thresholds \
 120                 write-detector \
 121                 sequence-test-detector | tee -a ${EXPERIMENT_RESULT_DIR}/stdout
 122
 123         else
 124
 125             echo "${EXPERIMENT_RESULT_DIR} exists, aborting experiment."
 126
 127         fi
 128
 129     done
 130
 131 done