projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
3d0bc1a
)
Fix the condition prefetch_needs_rand in the ImageDataLayer
author
Kai Li
<kaili_kloud@163.com>
Tue, 17 Jun 2014 03:43:10 +0000
(11:43 +0800)
committer
Kai Li
<kaili_kloud@163.com>
Fri, 20 Jun 2014 09:28:07 +0000
(17:28 +0800)
src/caffe/layers/image_data_layer.cpp
patch
|
blob
|
history
diff --git
a/src/caffe/layers/image_data_layer.cpp
b/src/caffe/layers/image_data_layer.cpp
index
9a79e72
..
1f7368e
100644
(file)
--- a/
src/caffe/layers/image_data_layer.cpp
+++ b/
src/caffe/layers/image_data_layer.cpp
@@
-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));