fix output path

tools/compress_model.py

@@ -78,6 +78,7 @@ def main():
     print('Compressed network prototxt {}'.format(args.prototxt_svd))
 
     out = os.path.splitext(os.path.basename(args.caffemodel))[0] + '_svd'
+    out_dir = os.path.dirname(args.caffemodel)
 
     # Compress fc6
     if net_svd.params.has_key('fc6_L'):
@@ -121,7 +122,7 @@ def main():
 
         out += '_fc7_{}'.format(l_fc7)
 
-    filename = 'snapshots/{}.caffemodel'.format(out)
+    filename = '{}/{}.caffemodel'.format(out_dir, out)
     net_svd.save(filename)
     print 'Wrote svd model to: {:s}'.format(filename)
 

tools/scripts/fetch_fast_rcnn_models.sh

+#!/bin/bash
+
+DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )/../../data" && pwd )"
+cd $DIR
+
+FILE=fast_rcnn_models.tgz
+# TODO
+URL=
+# TODO
+CHECKSUM=
+
+if [ -f $FILE ]; then
+  echo "File already exists. Checking md5..."
+  os=`uname -s`
+  if [ "$os" = "Linux" ]; then
+    checksum=`md5sum $FILE | awk '{ print $1 }'`
+  elif [ "$os" = "Darwin" ]; then
+    checksum=`cat $FILE | md5`
+  fi
+  if [ "$checksum" = "$CHECKSUM" ]; then
+    echo "Checksum is correct. No need to download."
+    exit 0
+  else
+    echo "Checksum is incorrect. Need to download again."
+  fi
+fi
+
+# TODO
+echo "Downloading precomputed selective search boxes (1.8G)..."
+
+wget $URL -O $FILE
+
+echo "Unzipping..."
+
+tar zxvf $FILE
+
+echo "Done. Please run this command again to verify that checksum = $CHECKSUM."