| 100 | |
| 101 | template <typename Dtype> |
| 102 | void ImageDataLayer<Dtype>::ShuffleImages() { |
| 103 | caffe::rng_t* prefetch_rng = |
| 104 | static_cast<caffe::rng_t*>(prefetch_rng_->generator()); |
| 105 | shuffle(lines_.begin(), lines_.end(), prefetch_rng); |
| 106 | } |
| 107 | |
| 108 | // This function is called on prefetch thread |
| 109 | template <typename Dtype> |