From bb5f9e67522064eeabf915188788bd27535f377a Mon Sep 17 00:00:00 2001 From: Kai Li Date: Tue, 17 Jun 2014 11:43:10 +0800 Subject: [PATCH] Fix the condition prefetch_needs_rand in the ImageDataLayer --- src/caffe/layers/image_data_layer.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/caffe/layers/image_data_layer.cpp b/src/caffe/layers/image_data_layer.cpp index 9a79e72..1f7368e 100644 --- a/src/caffe/layers/image_data_layer.cpp +++ b/src/caffe/layers/image_data_layer.cpp @@ -236,9 +236,7 @@ void ImageDataLayer::CreatePrefetchThread() { phase_ = Caffe::phase(); const bool prefetch_needs_rand = this->layer_param_.image_data_param().shuffle() || - ((phase_ == Caffe::TRAIN) && - (this->layer_param_.image_data_param().mirror() || - this->layer_param_.image_data_param().crop_size())); + this->layer_param_.image_data_param().crop_size(); if (prefetch_needs_rand) { const unsigned int prefetch_rng_seed = caffe_rng_rand(); prefetch_rng_.reset(new Caffe::RNG(prefetch_rng_seed)); -- 2.7.4