organizing experiments

potterhsu · Apr 21, 2015 · 5161451 · 5161451
1 parent 44c0412
commit 5161451
Show file tree

Hide file tree

Showing 14 changed files with 156 additions and 4 deletions.
diff --git a/experiments/fc_only.yml → experiments/cfgs/fc_only.yml b/experiments/fc_only.yml → experiments/cfgs/fc_only.yml
diff --git a/experiments/multiscale.yml → experiments/cfgs/multiscale.yml b/experiments/multiscale.yml → experiments/cfgs/multiscale.yml
diff --git a/experiments/no_bbox_reg.yml → experiments/cfgs/no_bbox_reg.yml b/experiments/no_bbox_reg.yml → experiments/cfgs/no_bbox_reg.yml
diff --git a/experiments/piecewise.yml → experiments/cfgs/piecewise.yml b/experiments/piecewise.yml → experiments/cfgs/piecewise.yml
diff --git a/experiments/svm.yml → experiments/cfgs/svm.yml b/experiments/svm.yml → experiments/cfgs/svm.yml
diff --git a/experiments/logs/.gitignore b/experiments/logs/.gitignore
@@ -0,0 +1 @@
+*.txt*
diff --git a/experiments/scripts/default_caffenet.sh b/experiments/scripts/default_caffenet.sh
@@ -0,0 +1,24 @@
+#! /bin/bash
+
+set -x
+
+export PYTHONUNBUFFERED="True"
+
+# -----------------------------------------------------------------------------
+LOG="experiments/logs/default_caffenet.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
+exec 3>&1 4>&2 &> >(tee -a "$LOG")
+echo Logging output to "$LOG"
+
+time ./tools/train_net.py --gpu $1 \
+  --solver models/CaffeNet/solver.prototxt \
+  --weights data/imagenet_models/CaffeNet.v2.caffemodel \
+  --imdb voc_2007_trainval
+
+time ./tools/test_net.py --gpu $1 \
+  --def models/CaffeNet/test.prototxt \
+  --net output/default/voc_2007_trainval/caffenet_fast_rcnn_iter_40000.caffemodel \
+  --imdb voc_2007_test
+
+# restore stdout/err
+exec 1>&3 2>&4
+# -----------------------------------------------------------------------------
diff --git a/experiments/scripts/default_vgg_cnn_m_1024.sh b/experiments/scripts/default_vgg_cnn_m_1024.sh
@@ -0,0 +1,24 @@
+#! /bin/bash
+
+set -x
+
+export PYTHONUNBUFFERED="True"
+
+# -----------------------------------------------------------------------------
+LOG="experiments/logs/default_vgg_cnn_m_1024.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
+exec 3>&1 4>&2 &> >(tee -a "$LOG")
+echo Logging output to "$LOG"
+
+time ./tools/train_net.py --gpu $1 \
+  --solver models/VGG_CNN_M_1024/solver.prototxt \
+  --weights data/imagenet_models/VGG_CNN_M_1024.v2.caffemodel \
+  --imdb voc_2007_trainval
+
+time ./tools/test_net.py --gpu $1 \
+  --def models/VGG_CNN_M_1024/test.prototxt \
+  --net output/default/voc_2007_trainval/vgg_cnn_m_1024_fast_rcnn_iter_40000.caffemodel \
+  --imdb voc_2007_test
+
+# restore stdout/err
+exec 1>&3 2>&4
+# -----------------------------------------------------------------------------
diff --git a/experiments/scripts/multiscale_caffenet.sh b/experiments/scripts/multiscale_caffenet.sh
@@ -0,0 +1,26 @@
+#! /bin/bash
+
+set -x
+
+export PYTHONUNBUFFERED="True"
+
+# -----------------------------------------------------------------------------
+LOG="experiments/logs/multiscale_caffenet.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
+exec 3>&1 4>&2 &> >(tee -a "$LOG")
+echo Logging output to "$LOG"
+
+time ./tools/train_net.py --gpu $1 \
+  --solver models/CaffeNet/solver.prototxt \
+  --weights data/imagenet_models/CaffeNet.v2.caffemodel \
+  --imdb voc_2007_trainval \
+  --cfg experiments/cfgs/multiscale.yml
+
+time ./tools/test_net.py --gpu $1 \
+  --def models/CaffeNet/test.prototxt \
+  --net output/multiscale/voc_2007_trainval/caffenet_fast_rcnn_multiscale_iter_40000.caffemodel \
+  --imdb voc_2007_test \
+  --cfg experiments/cfgs/multiscale.yml
+
+# restore stdout/err
+exec 1>&3 2>&4
+# -----------------------------------------------------------------------------
diff --git a/experiments/scripts/multiscale_vgg_cnn_m_1024.sh b/experiments/scripts/multiscale_vgg_cnn_m_1024.sh
@@ -0,0 +1,26 @@
+#! /bin/bash
+
+set -x
+
+export PYTHONUNBUFFERED="True"
+
+# -----------------------------------------------------------------------------
+LOG="experiments/logs/multiscale_vgg_cnn_m_1024.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
+exec 3>&1 4>&2 &> >(tee -a "$LOG")
+echo Logging output to "$LOG"
+
+time ./tools/train_net.py --gpu $1 \
+  --solver models/VGG_CNN_M_1024/solver.prototxt \
+  --weights data/imagenet_models/VGG_CNN_M_1024.v2.caffemodel \
+  --imdb voc_2007_trainval \
+  --cfg experiments/cfgs/multiscale.yml
+
+time ./tools/test_net.py --gpu $1 \
+  --def models/VGG_CNN_M_1024/test.prototxt \
+  --net output/multiscale/voc_2007_trainval/vgg_cnn_m_1024_fast_rcnn_multiscale_iter_40000.caffemodel \
+  --imdb voc_2007_test \
+  --cfg experiments/cfgs/multiscale.yml
+
+# restore stdout/err
+exec 1>&3 2>&4
+# -----------------------------------------------------------------------------
diff --git a/experiments/scripts/svd_caffenet.sh b/experiments/scripts/svd_caffenet.sh
@@ -0,0 +1,24 @@
+#! /bin/bash
+
+set -x
+
+export PYTHONUNBUFFERED="True"
+
+# -----------------------------------------------------------------------------
+LOG="experiments/logs/svd_caffenet.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
+exec 3>&1 4>&2 &> >(tee -a "$LOG")
+echo Logging output to "$LOG"
+
+time ./tools/compress_net.py \
+  --def models/CaffeNet/test.prototxt \
+  --def-svd models/CaffeNet/compressed/test.prototxt \
+  --net output/default/voc_2007_trainval/caffenet_fast_rcnn_iter_40000.caffemodel
+
+time ./tools/test_net.py --gpu $1 \
+  --def models/CaffeNet/compressed/test.prototxt \
+  --net output/default/voc_2007_trainval/caffenet_fast_rcnn_iter_40000_svd_fc6_1024_fc7_256.caffemodel \
+  --imdb voc_2007_test
+
+# restore stdout/err
+exec 1>&3 2>&4
+# -----------------------------------------------------------------------------
diff --git a/experiments/scripts/svd_vgg_cnn_m_1024.sh b/experiments/scripts/svd_vgg_cnn_m_1024.sh
@@ -0,0 +1,24 @@
+#! /bin/bash
+
+set -x
+
+export PYTHONUNBUFFERED="True"
+
+# -----------------------------------------------------------------------------
+LOG="experiments/logs/svd_vgg_cnn_m_1024.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
+exec 3>&1 4>&2 &> >(tee -a "$LOG")
+echo Logging output to "$LOG"
+
+time ./tools/compress_net.py \
+  --def models/VGG_CNN_M_1024/test.prototxt \
+  --def-svd models/VGG_CNN_M_1024/compressed/test.prototxt \
+  --net output/default/voc_2007_trainval/vgg_cnn_m_1024_fast_rcnn_iter_40000.caffemodel
+
+time ./tools/test_net.py --gpu $1 \
+  --def models/VGG_CNN_M_1024/compressed/test.prototxt \
+  --net output/default/voc_2007_trainval/vgg_cnn_m_1024_fast_rcnn_iter_40000_svd_fc6_1024_fc7_256.caffemodel \
+  --imdb voc_2007_test
+
+# restore stdout/err
+exec 1>&3 2>&4
+# -----------------------------------------------------------------------------
diff --git a/models/CaffeNet/compressed/test.prototxt b/models/CaffeNet/compressed/test.prototxt
@@ -27,6 +27,7 @@ layer {
   convolution_param {
     num_output: 96
     kernel_size: 11
+    pad: 5
     stride: 4
   }
 }
@@ -44,6 +45,7 @@ layer {
   pooling_param {
     pool: MAX
     kernel_size: 3
+    pad: 1
     stride: 2
   }
 }
@@ -73,8 +75,8 @@ layer {
   }
   convolution_param {
     num_output: 256
-    pad: 2
     kernel_size: 5
+    pad: 2
     group: 2
   }
 }
@@ -92,6 +94,7 @@ layer {
   pooling_param {
     pool: MAX
     kernel_size: 3
+    pad: 1
     stride: 2
   }
 }
@@ -121,8 +124,8 @@ layer {
   }
   convolution_param {
     num_output: 384
-    pad: 1
     kernel_size: 3
+    pad: 1
   }
 }
 layer {
@@ -146,8 +149,8 @@ layer {
   }
   convolution_param {
     num_output: 384
-    pad: 1
     kernel_size: 3
+    pad: 1
     group: 2
   }
 }
@@ -172,8 +175,8 @@ layer {
   }
   convolution_param {
     num_output: 256
-    pad: 1
     kernel_size: 3
+    pad: 1
     group: 2
   }
 }

diff --git a/tools/compress_model.py → tools/compress_net.py b/tools/compress_model.py → tools/compress_net.py