Fixed DetectionOutput layer

9 years ago · 7a2d08aa15
parent d392375c45
commit 7a2d08aa15
1 changed files with 1 additions and 1 deletions
--- a/modules/dnn/src/layers/detection_output_layer.cpp
+++ b/modules/dnn/src/layers/detection_output_layer.cpp
@ -297,7 +297,7 @@ void DetectionOutputLayer::forward(std::vector<Blob*> &inputs,
    std::vector<int> outputsShape(2, 1);
    outputsShape.push_back(numKept);
    outputsShape.push_back(7);
-    outputs[0].reshape(outputsShape);
+    outputs[0].create(outputsShape);
    float* outputsData = outputs[0].ptrf();

    int count = 0;