From 34cc67c6f295be909a7a1ad8525cd51ee22ac2a7 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Sun, 2 Aug 2020 21:11:39 +0800 Subject: [PATCH] fix exclude_nms (#1142) --- ppdet/modeling/architectures/cascade_mask_rcnn.py | 2 +- ppdet/modeling/architectures/cascade_rcnn_cls_aware.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ppdet/modeling/architectures/cascade_mask_rcnn.py b/ppdet/modeling/architectures/cascade_mask_rcnn.py index f353c7c18..97d9665d7 100644 --- a/ppdet/modeling/architectures/cascade_mask_rcnn.py +++ b/ppdet/modeling/architectures/cascade_mask_rcnn.py @@ -437,4 +437,4 @@ class CascadeMaskRCNN(object): def test(self, feed_vars, exclude_nms=False): assert not exclude_nms, "exclude_nms for {} is not support currently".format( self.__class__.__name__) - return self.build(feed_vars, 'test', exclude_nms=exclude_nms) + return self.build(feed_vars, 'test') diff --git a/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py b/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py index d905e199b..a8773f3a8 100644 --- a/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py +++ b/ppdet/modeling/architectures/cascade_rcnn_cls_aware.py @@ -319,7 +319,7 @@ class CascadeRCNNClsAware(object): return self.build_multi_scale(feed_vars) return self.build(feed_vars, 'test') - def test(self, feed_vars): + def test(self, feed_vars, exclude_nms=False): assert not exclude_nms, "exclude_nms for {} is not support currently".format( self.__class__.__name__) return self.build(feed_vars, 'test') -- GitLab