some cleanup

diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp
index 6225563..90e2caa 100644 (file)
--- a/include/caffe/vision_layers.hpp
+++ b/include/caffe/vision_layers.hpp
@@ -621,6 +621,43 @@ class AccuracyLayer : public Layer<Dtype> {
   }
 };
 
+// This function is used to create a pthread that prefetches the window data.
+template <typename Dtype>
+void* WindowDataLayerPrefetch(void* layer_pointer);
+
+template <typename Dtype>
+class WindowDataLayer : public Layer<Dtype> {
+  // The function used to perform prefetching.
+  friend void* WindowDataLayerPrefetch<Dtype>(void* layer_pointer);
+
+ public:
+  explicit WindowDataLayer(const LayerParameter& param)
+      : Layer<Dtype>(param) {}
+  virtual ~WindowDataLayer();
+  virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
+      vector<Blob<Dtype>*>* top);
+
+ protected:
+  virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
+      vector<Blob<Dtype>*>* top);
+  virtual void Forward_gpu(const vector<Blob<Dtype>*>& bottom,
+      vector<Blob<Dtype>*>* top);
+  virtual Dtype Backward_cpu(const vector<Blob<Dtype>*>& top,
+      const bool propagate_down, vector<Blob<Dtype>*>* bottom);
+  virtual Dtype Backward_gpu(const vector<Blob<Dtype>*>& top,
+      const bool propagate_down, vector<Blob<Dtype>*>* bottom);
+
+  pthread_t thread_;
+  shared_ptr<Blob<Dtype> > prefetch_data_;
+  shared_ptr<Blob<Dtype> > prefetch_label_;
+  Blob<Dtype> data_mean_;
+  vector<std::pair<std::string, vector<int> > > image_database_;
+  enum WindowField { IMAGE_INDEX, LABEL, OVERLAP, X1, Y1, X2, Y2, NUM };
+  vector<vector<float> > fg_windows_;
+  vector<vector<float> > bg_windows_;
+};
+
+
 }  // namespace caffe
 
 #endif  // CAFFE_VISION_LAYERS_HPP_

diff --git a/models/pascal_finetune.prototxt b/models/pascal_finetune.prototxt
index c2875c0..fc2085e 100644 (file)
--- a/models/pascal_finetune.prototxt
+++ b/models/pascal_finetune.prototxt
@@ -7,9 +7,9 @@ layers {
     meanfile: "/home/eecs/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
     batchsize: 128
     cropsize: 227
-    context_pad: 16
-    crop_mode: "warp"
     mirror: true
+    det_context_pad: 16
+    det_crop_mode: "warp"
     det_fg_threshold: 0.5
     det_bg_threshold: 0.5
     det_fg_fraction: 0.25
@@ -340,7 +340,7 @@ layers {
 }
 layers {
   layer {
-    name: "fc8"
+    name: "fc8_pascal"
     type: "innerproduct"
     num_output: 21
     weight_filler {
@@ -355,16 +355,15 @@ layers {
     blobs_lr: 20.
     weight_decay: 1.
     weight_decay: 0.
-    can_clobber: false
   }
   bottom: "fc7"
-  top: "fc8"
+  top: "fc8_pascal"
 }
 layers {
   layer {
     name: "loss"
     type: "softmax_loss"
   }
-  bottom: "fc8"
+  bottom: "fc8_pascal"
   bottom: "label"
 }

diff --git a/models/pascal_finetune_solver.prototxt b/models/pascal_finetune_solver.prototxt
index cbdef7f..ee43c40 100644 (file)
--- a/models/pascal_finetune_solver.prototxt
+++ b/models/pascal_finetune_solver.prototxt
@@ -11,4 +11,4 @@ max_iter: 100000
 momentum: 0.9
 weight_decay: 0.0005
 snapshot: 10000
-snapshot_prefix: "./snapshots/pascal_warp_context16_finetune_train"
+snapshot_prefix: "./snapshots/pascal_finetune_train"

diff --git a/models/pascal_finetune_val.prototxt b/models/pascal_finetune_val.prototxt
index 0c47459..6453431 100644 (file)
--- a/models/pascal_finetune_val.prototxt
+++ b/models/pascal_finetune_val.prototxt
@@ -7,9 +7,9 @@ layers {
     meanfile: "/home/eecs/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
     batchsize: 128
     cropsize: 227
-    context_pad: 16
-    crop_mode: "warp"
     mirror: true
+    det_context_pad: 16
+    det_crop_mode: "warp"
     det_fg_threshold: 0.5
     det_bg_threshold: 0.5
     det_fg_fraction: 0.25
@@ -340,7 +340,7 @@ layers {
 }
 layers {
   layer {
-    name: "fc8"
+    name: "fc8_pascal"
     type: "innerproduct"
     num_output: 21
     weight_filler {
@@ -357,14 +357,14 @@ layers {
     weight_decay: 0.
   }
   bottom: "fc7"
-  top: "fc8"
+  top: "fc8_pascal"
 }
 layers {
   layer {
     name: "prob"
     type: "softmax"
   }
-  bottom: "fc8"
+  bottom: "fc8_pascal"
   top: "prob"
 }
 layers {

diff --git a/src/caffe/layers/window_data_layer.cpp b/src/caffe/layers/window_data_layer.cpp
index 2dae2f9..3fdc4f9 100644 (file)
--- a/src/caffe/layers/window_data_layer.cpp
+++ b/src/caffe/layers/window_data_layer.cpp
@@ -1,10 +1,13 @@
 // Copyright 2013 Ross Girshick
+//
+// Based on data_layer.cpp by Yangqing Jia.
 
 #include <stdint.h>
 #include <pthread.h>
 
 #include <string>
 #include <vector>
+#include <map>
 #include <fstream>
 
 #include "caffe/layer.hpp"
@@ -16,14 +19,13 @@
 #include <opencv2/imgproc/imgproc.hpp>
 
 using std::string;
+using std::map;
+using std::pair;
 
 // caffe.proto > LayerParameter
 //   'source' field specifies the window_file
 //   'cropsize' indicates the desired warped size
 
-// TODO(rbg):
-//  - try uniform sampling over classes
-
 namespace caffe {
 
 template <typename Dtype>
@@ -39,7 +41,7 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
   const Dtype scale = layer->layer_param_.scale();
   const int batchsize = layer->layer_param_.batchsize();
   const int cropsize = layer->layer_param_.cropsize();
-  const int context_pad = layer->layer_param_.context_pad();
+  const int context_pad = layer->layer_param_.det_context_pad();
   const bool mirror = layer->layer_param_.mirror();
   const float fg_fraction = layer->layer_param_.det_fg_fraction();
   const Dtype* mean = layer->data_mean_.cpu_data();
@@ -47,17 +49,13 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
   const int mean_width = layer->data_mean_.width();
   const int mean_height = layer->data_mean_.height();
   cv::Size cv_crop_size(cropsize, cropsize);
-  const string& crop_mode = layer->layer_param_.crop_mode();
+  const string& crop_mode = layer->layer_param_.det_crop_mode();
 
   bool use_square = (crop_mode == "square") ? true : false;
 
   // zero out batch
   memset(top_data, 0, sizeof(Dtype)*layer->prefetch_data_->count());
 
-//  CHECK_EQ(mean_width, mean_height);
-//  CHECK_EQ(mean_width, 256);
-//  CHECK_EQ(mean_off, 14);
-
   const int num_fg = static_cast<int>(static_cast<float>(batchsize) 
       * fg_fraction);
   const int num_samples[2] = { batchsize - num_fg, num_fg };
@@ -77,7 +75,7 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
       }
 
       // load the image containing the window
-      std::pair<std::string, vector<int> > image = 
+      pair<std::string, vector<int> > image = 
           layer->image_database_[window[WindowDataLayer<Dtype>::IMAGE_INDEX]];
 
       cv::Mat cv_img = cv::imread(image.first, CV_LOAD_IMAGE_COLOR);
@@ -86,7 +84,6 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
         return (void*)NULL;
       }
       const int channels = cv_img.channels();
-//      CHECK_EQ(channels, 3);
 
       // crop window out of image and warp it
       int x1 = window[WindowDataLayer<Dtype>::X1];
@@ -177,15 +174,6 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
         }
       }
 
-//      CHECK_GT(x1, -1);
-//      CHECK_GT(y1, -1);
-//      CHECK_LT(x1, cv_img.cols);
-//      CHECK_LT(y1, cv_img.rows);
-//      CHECK_GT(x2, x1-1);
-//      CHECK_GT(y2, y1-1);
-//      CHECK_LT(x2, cv_img.cols);
-//      CHECK_LT(y2, cv_img.rows);
-
       cv::Rect roi(x1, y1, x2-x1+1, y2-y1+1);
       cv::Mat cv_cropped_img = cv_img(roi);
       cv::resize(cv_cropped_img, cv_cropped_img, 
@@ -215,33 +203,38 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
       // get window label
       top_label[itemid] = window[WindowDataLayer<Dtype>::LABEL];
 
-//      string file_id;
-//      std::stringstream ss;
-//      ss << rand();
-//      ss >> file_id;
-//      std::ofstream inf((string("dump/") + file_id + string("_info.txt")).c_str(), std::ofstream::out);
-//      inf << image.first << std::endl 
-//          << window[WindowDataLayer<Dtype>::X1]+1 << std::endl
-//          << window[WindowDataLayer<Dtype>::Y1]+1 << std::endl
-//          << window[WindowDataLayer<Dtype>::X2]+1 << std::endl
-//          << window[WindowDataLayer<Dtype>::Y2]+1 << std::endl
-//          << do_mirror << std::endl
-//          << top_label[itemid] << std::endl
-//          << is_fg << std::endl;
-//      inf.close();
-//      std::ofstream top_data_file((string("dump/") + file_id + string("_data.txt")).c_str(), 
-//            std::ofstream::out | std::ofstream::binary);
-//      for (int c = 0; c < channels; ++c) {
-//        for (int h = 0; h < cropsize; ++h) {
-//          for (int w = 0; w < cropsize; ++w) {
-//            top_data_file.write(
-//                reinterpret_cast<char*>(&top_data[((itemid * channels + c) 
-//                                                   * cropsize + h) * cropsize + w]),
-//                sizeof(Dtype));
-//          }
-//        }
-//      }
-//      top_data_file.close();
+      #if 0
+      // useful debugging code for dumping transformed windows to disk
+      string file_id;
+      std::stringstream ss;
+      ss << rand();
+      ss >> file_id;
+      std::ofstream inf((string("dump/") + file_id + 
+          string("_info.txt")).c_str(), std::ofstream::out);
+      inf << image.first << std::endl 
+          << window[WindowDataLayer<Dtype>::X1]+1 << std::endl
+          << window[WindowDataLayer<Dtype>::Y1]+1 << std::endl
+          << window[WindowDataLayer<Dtype>::X2]+1 << std::endl
+          << window[WindowDataLayer<Dtype>::Y2]+1 << std::endl
+          << do_mirror << std::endl
+          << top_label[itemid] << std::endl
+          << is_fg << std::endl;
+      inf.close();
+      std::ofstream top_data_file((string("dump/") + file_id + 
+          string("_data.txt")).c_str(), 
+          std::ofstream::out | std::ofstream::binary);
+      for (int c = 0; c < channels; ++c) {
+        for (int h = 0; h < cropsize; ++h) {
+          for (int w = 0; w < cropsize; ++w) {
+            top_data_file.write(
+                reinterpret_cast<char*>(&top_data[((itemid * channels + c) 
+                                                   * cropsize + h) * cropsize + w]),
+                sizeof(Dtype));
+          }
+        }
+      }
+      top_data_file.close();
+      #endif
 
       itemid++;
     }
@@ -250,6 +243,10 @@ void* WindowDataLayerPrefetch(void* layer_pointer) {
   return (void*)NULL;
 }
 
+template <typename Dtype>
+WindowDataLayer<Dtype>::~WindowDataLayer<Dtype>() {
+  CHECK(!pthread_join(thread_, NULL)) << "Pthread joining failed.";
+}
 
 template <typename Dtype>
 void WindowDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
@@ -284,8 +281,8 @@ void WindowDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
   CHECK(infile.good()) << "Failed to open window file " 
       << this->layer_param_.source() << std::endl;
 
-  vector<float> label_hist(21);
-  std::fill(label_hist.begin(), label_hist.end(), 0);
+  map<int, int> label_hist;
+  label_hist.insert(std::make_pair(0, 0));
 
   string hashtag;
   int image_index, channels;
@@ -319,15 +316,18 @@ void WindowDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
       
       // add window to foreground list or background list
       if (overlap >= this->layer_param_.det_fg_threshold()) {
-        CHECK_GT(window[WindowDataLayer::LABEL], 0);
+        int label = window[WindowDataLayer::LABEL];
+        CHECK_GT(label, 0);
         fg_windows_.push_back(window);
+        label_hist.insert(std::make_pair(label, 0));
+        label_hist[label]++;
       } else if (overlap < this->layer_param_.det_bg_threshold()) {
         // background window, force label and overlap to 0
         window[WindowDataLayer::LABEL] = 0;
         window[WindowDataLayer::OVERLAP] = 0;
         bg_windows_.push_back(window);
+        label_hist[0]++;
       }
-      label_hist[window[WindowDataLayer::LABEL]]++;
     }
 
     if (image_index % 100 == 0) {
@@ -342,14 +342,15 @@ void WindowDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
 
   LOG(INFO) << "Number of images: " << image_index+1;
 
-  for (int i = 0; i < 21; ++i) {
-    LOG(INFO) << "class " << i << " has " << label_hist[i] << " samples";
+  for (map<int,int>::iterator it = label_hist.begin(); 
+      it != label_hist.end(); ++it) {
+    LOG(INFO) << "class " << it->first << " has " << label_hist[it->first] << " samples";
   }
 
   LOG(INFO) << "Amount of context padding: " 
-      << this->layer_param_.context_pad();
+      << this->layer_param_.det_context_pad();
 
-  LOG(INFO) << "Crop mode: " << this->layer_param_.crop_mode();
+  LOG(INFO) << "Crop mode: " << this->layer_param_.det_crop_mode();
 
   // image
   int cropsize = this->layer_param_.cropsize();

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index 1837b07..af33770 100644 (file)
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -273,10 +273,10 @@ void Net<Dtype>::CopyTrainedLayersFrom(const NetParameter& param) {
       ++target_layer_id;
     }
     if (target_layer_id == layer_names_.size()) {
-      DLOG(INFO) << "Ignoring source layer " << source_layer_name;
+      LOG(INFO) << "Ignoring source layer " << source_layer_name;
       continue;
     }
-    DLOG(INFO) << "Copying source layer " << source_layer_name;
+    LOG(INFO) << "Copying source layer " << source_layer_name;
     vector<shared_ptr<Blob<Dtype> > >& target_blobs =
         layers_[target_layer_id]->blobs();
     CHECK_EQ(target_blobs.size(), source_layer.blobs_size())

diff --git a/src/caffe/proto/caffe.proto b/src/caffe/proto/caffe.proto
index b86e5f3..406f5eb 100644 (file)
--- a/src/caffe/proto/caffe.proto
+++ b/src/caffe/proto/caffe.proto
@@ -100,18 +100,16 @@ message LayerParameter {
   // Fraction of batch that should be foreground objects
   optional float det_fg_fraction = 56 [default = 0.25];
 
-  // Set can_clobber to false to avoid overriding it with a pretrained
-  // network's parameters when finetuning
-  optional bool can_clobber = 57 [default = true];
+  // optional bool OBSOLETE_can_clobber = 57 [ default = true ];
 
   // Amount of contextual padding to add around a window
   // (used only by the window_data_layer)
-  optional uint32 context_pad = 58 [default = 0];
+  optional uint32 det_context_pad = 58 [default = 0];
 
   // Mode for cropping out a detection window
   // warp: cropped window is warped to a fixed size and aspect ratio
   // square: the tightest square around the window is cropped
-  optional string crop_mode = 59 [default = "warp"];
+  optional string det_crop_mode = 59 [default = "warp"];
 
   // For ReshapeLayer, one needs to specify the new dimensions.
   optional int32 new_num = 60 [default = 0];