Skip to content

Commit f7175f5

Browse files
committed
dnn: fix additional text boxes handling after the latest adaptations for TF
1 parent 050916f commit f7175f5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

modules/dnn/src/layers/prior_box_layer.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,10 +254,11 @@ class PriorBoxLayerImpl : public PriorBoxLayer
254254
CV_TRACE_FUNCTION();
255255
CV_TRACE_ARG_VALUE(name, "name", name.c_str());
256256

257+
size_t real_numPriors = _additional_y_offset ? _numPriors / 2 : _numPriors;
257258
if (_scales.empty())
258-
_scales.resize(_numPriors, 1.0f);
259+
_scales.resize(real_numPriors, 1.0f);
259260
else
260-
CV_Assert(_scales.size() == _numPriors);
261+
CV_Assert(_scales.size() == real_numPriors);
261262

262263
int _layerWidth = inputs[0]->size[3];
263264
int _layerHeight = inputs[0]->size[2];

0 commit comments

Comments
 (0)