未验证 提交 1e52b710 编写于 作者: W wangxinxin08 提交者: GitHub

modify infer problem of s2anet (#5326)

上级 58f9a903
...@@ -251,7 +251,7 @@ class S2ANetBBoxPostProcess(nn.Layer): ...@@ -251,7 +251,7 @@ class S2ANetBBoxPostProcess(nn.Layer):
def __init__(self, num_classes=15, nms_pre=2000, min_bbox_size=0, nms=None): def __init__(self, num_classes=15, nms_pre=2000, min_bbox_size=0, nms=None):
super(S2ANetBBoxPostProcess, self).__init__() super(S2ANetBBoxPostProcess, self).__init__()
self.num_classes = num_classes self.num_classes = num_classes
self.nms_pre = paddle.to_tensor(nms_pre) self.nms_pre = nms_pre
self.min_bbox_size = min_bbox_size self.min_bbox_size = min_bbox_size
self.nms = nms self.nms = nms
self.origin_shape_list = [] self.origin_shape_list = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册