зеркало из https://github.com/microsoft/caffe.git
Merge pull request #805 from alfredtofu/dev
Update image_data_layer.cpp
This commit is contained in:
Коммит
86cc3e91ae
|
@ -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.";
|
||||
|
|
Загрузка…
Ссылка в новой задаче