提交 71b3fbb1 编写于 作者: G gaoyuan

Fix a ssd bug

上级 462c2ed8
...@@ -143,7 +143,8 @@ void DetectionOutputLayer::forward(PassType passType) { ...@@ -143,7 +143,8 @@ 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.
先完成此消息的编辑!
想要评论请 注册