File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -852,16 +852,16 @@ class DetectionOutputLayerImpl CV_FINAL : public DetectionOutputLayer
852
852
for (int i = 0 ; i < num; ++i, locData += numPredsPerClass * numLocClasses * 4 )
853
853
{
854
854
LabelBBox& labelBBox = locPreds[i];
855
+ int start = shareLocation ? -1 : 0 ;
856
+ for (int c = 0 ; c < numLocClasses; ++c) {
857
+ labelBBox[start++].resize (numPredsPerClass);
858
+ }
855
859
for (int p = 0 ; p < numPredsPerClass; ++p)
856
860
{
857
861
int startIdx = p * numLocClasses * 4 ;
858
862
for (int c = 0 ; c < numLocClasses; ++c)
859
863
{
860
864
int label = shareLocation ? -1 : c;
861
- if (labelBBox.find (label) == labelBBox.end ())
862
- {
863
- labelBBox[label].resize (numPredsPerClass);
864
- }
865
865
util::NormalizedBBox& bbox = labelBBox[label][p];
866
866
if (locPredTransposed)
867
867
{
You can’t perform that action at this time.
0 commit comments