From f7175f505057aa9c6adcb4edbcc2915d15c3b944 Mon Sep 17 00:00:00 2001 From: Vladislav Sovrasov Date: Wed, 11 Oct 2017 14:04:48 +0300 Subject: [PATCH] dnn: fix additional text boxes handling after the latest adaptations for TF --- modules/dnn/src/layers/prior_box_layer.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/layers/prior_box_layer.cpp b/modules/dnn/src/layers/prior_box_layer.cpp index 3f74472f40..3ca083503b 100644 --- a/modules/dnn/src/layers/prior_box_layer.cpp +++ b/modules/dnn/src/layers/prior_box_layer.cpp @@ -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];