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,
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;
......
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