зеркало из https://github.com/microsoft/caffe.git
Update image_data_layer.cpp
fix bug for getting new width
This commit is contained in:
Родитель
0e3d9701b7
Коммит
4f12cc7391
|
@ -138,7 +138,7 @@ void ImageDataLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom,
|
||||||
vector<Blob<Dtype>*>* top) {
|
vector<Blob<Dtype>*>* top) {
|
||||||
Layer<Dtype>::SetUp(bottom, top);
|
Layer<Dtype>::SetUp(bottom, top);
|
||||||
const int new_height = this->layer_param_.image_data_param().new_height();
|
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) ||
|
CHECK((new_height == 0 && new_width == 0) ||
|
||||||
(new_height > 0 && new_width > 0)) << "Current implementation requires "
|
(new_height > 0 && new_width > 0)) << "Current implementation requires "
|
||||||
"new_height and new_width to be set at the same time.";
|
"new_height and new_width to be set at the same time.";
|
||||||
|
|
Загрузка…
Ссылка в новой задаче