Merge pull request #805 from alfredtofu/dev

Update image_data_layer.cpp
This commit is contained in:
Yangqing Jia 2014-07-28 08:12:18 -07:00
Родитель f8f4294e26 4f12cc7391
Коммит 86cc3e91ae
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -138,7 +138,7 @@ void ImageDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top) {
Layer<Dtype>::SetUp(bottom, top);
const int new_height = this->layer_param_.image_data_param().new_height();
const int new_width = this->layer_param_.image_data_param().new_height();
const int new_width = this->layer_param_.image_data_param().new_width();
CHECK((new_height == 0 && new_width == 0) ||
(new_height > 0 && new_width > 0)) << "Current implementation requires "
"new_height and new_width to be set at the same time.";