Commit f7175f50 authored by Vladislav Sovrasov's avatar Vladislav Sovrasov

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

parent 050916fd
......@@ -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];
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment