Code that was used to finetune with reasonable success

diff --git a/models/pascal_finetune.prototxt b/models/pascal_finetune.prototxt
index 1ff9dab72d0a8244ec3fe8650377c8b690c7efb3..5ee9e8a9a5a686ac2c22d889055e395750986e43 100644 (file)
--- a/models/pascal_finetune.prototxt
+++ b/models/pascal_finetune.prototxt
@@ -3,7 +3,7 @@ layers {
   layer {
     name: "data"
     type: "window_data"
-    source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_train.txt"
+    source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_trainval.txt"
     meanfile: "/home/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
     batchsize: 128
     cropsize: 227

diff --git a/models/pascal_finetune_solver.prototxt b/models/pascal_finetune_solver.prototxt
index 14925ade811a464d0d33c88011034dfd80f9c09f..7ea9c5d4d59916ff312f683742f4d35977d3b0bc 100644 (file)
--- a/models/pascal_finetune_solver.prototxt
+++ b/models/pascal_finetune_solver.prototxt
@@ -1,13 +1,13 @@
 train_net: "examples/pascal_finetune.prototxt"
 test_net: "examples/pascal_finetune_val.prototxt"
 test_iter: 100
-test_interval: 500
+test_interval: 1000
 base_lr: 0.0001
 lr_policy: "step"
 gamma: 0.1
-stepsize: 10000
+stepsize: 20000
 display: 20
-max_iter: 400000
+max_iter: 100000
 momentum: 0.9
 weight_decay: 0.0005
 snapshot: 10000

diff --git a/models/pascal_finetune_val.prototxt b/models/pascal_finetune_val.prototxt
index 1cea488218e092202bf1b52d6259ca76bd729f5a..c73cf8f4a401965cf31fdbfe2556993e062db5df 100644 (file)
--- a/models/pascal_finetune_val.prototxt
+++ b/models/pascal_finetune_val.prototxt
@@ -3,7 +3,7 @@ layers {
   layer {
     name: "data"
     type: "window_data"
-    source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_val.txt"
+    source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_test.txt"
     meanfile: "/home/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
     batchsize: 128
     cropsize: 227

diff --git a/src/caffe/layers/window_data_layer.cpp b/src/caffe/layers/window_data_layer.cpp
index 366ae0bf731316fbda88efbb69061e8dee562022..a2346bff75c9ba41a50944c2b646e090545a0fd7 100644 (file)
--- a/src/caffe/layers/window_data_layer.cpp
+++ b/src/caffe/layers/window_data_layer.cpp
@@ -233,7 +233,7 @@ void WindowDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
       label_hist[window[WindowDataLayer::LABEL]]++;
     }
 
-    if (image_index % 1000 == 0) {
+    if (image_index % 1 == 0) {
       LOG(INFO) << "num: " << image_index << " "
           << image_path << " " 
           << image_size[0] << " "
@@ -243,6 +243,8 @@ void WindowDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
     }
   }
 
+  LOG(INFO) << "Number of images: " << image_index;
+
   for (int i = 0; i < 21; ++i) {
     LOG(INFO) << "class " << i << " has " << label_hist[i] << " samples";
   }