提交 6e9337e3 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #4182 from Noplz/develop

Fix a ssd bug
...@@ -143,7 +143,7 @@ void DetectionOutputLayer::forward(PassType passType) { ...@@ -143,7 +143,7 @@ void DetectionOutputLayer::forward(PassType passType) {
resetOutput(numKept, 7); resetOutput(numKept, 7);
} else { } else {
MatrixPtr outV = getOutputValue(); MatrixPtr outV = getOutputValue();
outV = NULL; if (outV) outV->resize(0, 0);
return; return;
} }
MatrixPtr outV = getOutputValue(); MatrixPtr outV = getOutputValue();
......
...@@ -1224,8 +1224,8 @@ def detection_output_layer(input_loc, ...@@ -1224,8 +1224,8 @@ def detection_output_layer(input_loc,
name=None): name=None):
""" """
Apply the NMS to the output of network and compute the predict bounding Apply the NMS to the output of network and compute the predict bounding
box location. The output of this layer could be None if there is no valid box location. The output's shape of this layer could be zero if there is
bounding box. no valid bounding box.
:param name: The Layer Name. :param name: The Layer Name.
:type name: basestring :type name: basestring
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册