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

fix Mask-RCNN QAT bug (#5690)

上级 f70a80fc
......@@ -33,7 +33,7 @@ __all__ = [
@register
class BBoxPostProcess(nn.Layer):
class BBoxPostProcess(object):
__shared__ = ['num_classes', 'export_onnx']
__inject__ = ['decode', 'nms']
......@@ -45,7 +45,7 @@ class BBoxPostProcess(nn.Layer):
self.nms = nms
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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册