Commit 7a2d08aa authored by Anna Petrovicheva's avatar Anna Petrovicheva

Fixed DetectionOutput layer

parent d392375c
...@@ -297,7 +297,7 @@ void DetectionOutputLayer::forward(std::vector<Blob*> &inputs, ...@@ -297,7 +297,7 @@ void DetectionOutputLayer::forward(std::vector<Blob*> &inputs,
std::vector<int> outputsShape(2, 1); std::vector<int> outputsShape(2, 1);
outputsShape.push_back(numKept); outputsShape.push_back(numKept);
outputsShape.push_back(7); outputsShape.push_back(7);
outputs[0].reshape(outputsShape); outputs[0].create(outputsShape);
float* outputsData = outputs[0].ptrf(); float* outputsData = outputs[0].ptrf();
int count = 0; int count = 0;
......
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