Merge pull request BVLC#1457 from jyegerlehner/preserve-extracted-blo…

…b-shapes extract_features preserves feature shape
anshumang · Mar 8, 2015 · c942dc1 · c942dc1
2 parents a9bf7b9 + 1cd6fcb
commit c942dc1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/extract_features.cpp b/tools/extract_features.cpp
@@ -147,9 +147,9 @@ int feature_extraction_pipeline(int argc, char** argv) {
       int dim_features = feature_blob->count() / batch_size;
       const Dtype* feature_blob_data;
       for (int n = 0; n < batch_size; ++n) {
-        datum.set_height(dim_features);
-        datum.set_width(1);
-        datum.set_channels(1);
+        datum.set_height(feature_blob->height());
+        datum.set_width(feature_blob->width());
+        datum.set_channels(feature_blob->channels());
         datum.clear_data();
         datum.clear_float_data();
         feature_blob_data = feature_blob->cpu_data() +