diff options
| author | Sergio <sguada@gmail.com> | 2014-10-15 15:36:42 -0700 |
|---|---|---|
| committer | Sergio <sguada@gmail.com> | 2014-10-15 17:03:18 -0700 |
| commit | 65688bed17b3bc162fd5a0db9c97de11c9131258 (patch) | |
| tree | d5b1506abd02a8fb70b1539edf7d50179f98c0d9 /src/caffe | |
| parent | d6b71e58237968679cea44418b869721a5ea9654 (diff) | |
| download | caffeonacl-65688bed17b3bc162fd5a0db9c97de11c9131258.tar.gz caffeonacl-65688bed17b3bc162fd5a0db9c97de11c9131258.tar.bz2 caffeonacl-65688bed17b3bc162fd5a0db9c97de11c9131258.zip | |
Upgrade compute_image_mean to use gflags, accept list_of_images, and print mean_values
Diffstat (limited to 'src/caffe')
| -rw-r--r-- | src/caffe/layers/data_layer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/layers/data_layer.cpp b/src/caffe/layers/data_layer.cpp index 7ddd4662..05f2e298 100644 --- a/src/caffe/layers/data_layer.cpp +++ b/src/caffe/layers/data_layer.cpp @@ -46,9 +46,9 @@ void DataLayer<Dtype>::DataLayerSetUp(const vector<Blob<Dtype>*>& bottom, } // Read a data point, and use it to initialize the top blob. CHECK(iter_ != dataset_->end()); - const Datum& datum = iter_->value; + Datum datum = iter_->value; - if (DecodeDatum(datum)) { + if (DecodeDatum(&datum)) { LOG(INFO) << "Decoding Datum"; } // image |