未验证 提交 34cc67c6 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix exclude_nms (#1142)

上级 d4afef30
......@@ -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')
......@@ -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')
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册