Fix a ssd bug

update-doc-pybind
gaoyuan 8 years ago
parent 462c2ed8bf
commit 71b3fbb18a

@ -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,
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

Loading…
Cancel
Save