fix bug for getting new width
This commit is contained in:
alfredtofu 2014-07-28 10:19:51 +08:00
Родитель 0e3d9701b7
Коммит 4f12cc7391
1 изменённых файлов: 1 добавлений и 1 удалений

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

@ -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.";