dnn: fix additional text boxes handling after the latest adaptations for TF

This commit is contained in:
Vladislav Sovrasov 2017-10-11 14:04:48 +03:00
parent 050916fd6b
commit f7175f5050

View File

@ -254,10 +254,11 @@ public:
CV_TRACE_FUNCTION();
CV_TRACE_ARG_VALUE(name, "name", name.c_str());
size_t real_numPriors = _additional_y_offset ? _numPriors / 2 : _numPriors;
if (_scales.empty())
_scales.resize(_numPriors, 1.0f);
_scales.resize(real_numPriors, 1.0f);
else
CV_Assert(_scales.size() == _numPriors);
CV_Assert(_scales.size() == real_numPriors);
int _layerWidth = inputs[0]->size[3];
int _layerHeight = inputs[0]->size[2];