未验证 提交 467873ad 编写于 作者: W wangguanzhong 提交者: GitHub

minor fix (#1241)

上级 c1c537b7
......@@ -2061,6 +2061,7 @@ class BboxXYXY2XYWH(BaseOperator):
return sample
@register_op
class Lighting(BaseOperator):
"""
Lighting the imagen by eigenvalues and eigenvectors
......
......@@ -47,7 +47,7 @@ class TTFNet(object):
self.ttf_head = ttf_head
self.num_classes = num_classes
def build(self, feed_vars, mode='train'):
def build(self, feed_vars, mode='train', exclude_nms=False):
im = feed_vars['image']
mixed_precision_enabled = mixed_precision_global_state() is not None
......@@ -128,5 +128,5 @@ class TTFNet(object):
def eval(self, feed_vars):
return self.build(feed_vars, mode='test')
def test(self, feed_vars):
return self.build(feed_vars, mode='test')
def test(self, feed_vars, exclude_nms=False):
return self.build(feed_vars, mode='test', exclude_nms=exclude_nms)
......@@ -125,7 +125,7 @@ def DeformConvNorm(input,
initializer=None,
bias_attr=False,
name=None):
assert norm_type in ['bn', 'sync_bn', 'affine_channel']
assert norm_type in ['bn', 'sync_bn', 'affine_channel', 'gn']
conv = DeformConv(input, num_filters, filter_size, stride, groups, dilation,
lr_scale, initializer, bias_attr, name)
......
......@@ -112,6 +112,8 @@ def dump_infer_config(FLAGS, config):
'RetinaNet': 40,
'Face': 3,
'TTFNet': 3,
'FCOS': 3,
'EfficientDet': 40
}
infer_arch = config['architecture']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册