未验证 提交 e366e929 编写于 作者: G Guanghua Yu 提交者: GitHub

fix Mask-RCNN QAT bug (#5690)

上级 f70a80fc
...@@ -33,7 +33,7 @@ __all__ = [ ...@@ -33,7 +33,7 @@ __all__ = [
@register @register
class BBoxPostProcess(nn.Layer): class BBoxPostProcess(object):
__shared__ = ['num_classes', 'export_onnx'] __shared__ = ['num_classes', 'export_onnx']
__inject__ = ['decode', 'nms'] __inject__ = ['decode', 'nms']
...@@ -45,9 +45,9 @@ class BBoxPostProcess(nn.Layer): ...@@ -45,9 +45,9 @@ class BBoxPostProcess(nn.Layer):
self.nms = nms self.nms = nms
self.export_onnx = export_onnx self.export_onnx = export_onnx
def forward(self, head_out, rois, im_shape, scale_factor): def __call__(self, head_out, rois, im_shape, scale_factor):
""" """
Decode the bbox and do NMS if needed. Decode the bbox and do NMS if needed.
Args: Args:
head_out (tuple): bbox_pred and cls_prob of bbox_head output. head_out (tuple): bbox_pred and cls_prob of bbox_head output.
...@@ -85,7 +85,7 @@ class BBoxPostProcess(nn.Layer): ...@@ -85,7 +85,7 @@ class BBoxPostProcess(nn.Layer):
""" """
Rescale, clip and filter the bbox from the output of NMS to Rescale, clip and filter the bbox from the output of NMS to
get final prediction. get final prediction.
Notes: Notes:
Currently only support bs = 1. Currently only support bs = 1.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册