diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py index 765e05fbf732a4d17d9b441ed355b4803ca64ab0..967d43db09bbb5660687b32919abc9eb5734ae7d 100644 --- a/ppdet/modeling/post_process.py +++ b/ppdet/modeling/post_process.py @@ -252,7 +252,7 @@ class S2ANetBBoxPostProcess(nn.Layer): def __init__(self, num_classes=15, nms_pre=2000, min_bbox_size=0, nms=None): super(S2ANetBBoxPostProcess, self).__init__() 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.nms = nms self.origin_shape_list = []