Fix the condition prefetch_needs_rand in the ImageDataLayer
authorKai Li <kaili_kloud@163.com>
Tue, 17 Jun 2014 03:43:10 +0000 (11:43 +0800)
committerKai Li <kaili_kloud@163.com>
Fri, 20 Jun 2014 09:28:07 +0000 (17:28 +0800)
src/caffe/layers/image_data_layer.cpp

index 9a79e72..1f7368e 100644 (file)
@@ -236,9 +236,7 @@ void ImageDataLayer<Dtype>::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));