minor fix (open-mmlab#1241)

ppdet/data/transform/operators.py

@@ -2061,6 +2061,7 @@ def __call__(self, sample, context=None):
         return sample
 
 
+@register_op
 class Lighting(BaseOperator):
     """
     Lighting the imagen by eigenvalues and eigenvectors

ppdet/modeling/architectures/ttfnet.py

@@ -47,7 +47,7 @@ def __init__(self, backbone, ttf_head='TTFHead', num_classes=80):
         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 @@ def train(self, feed_vars):
     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)

ppdet/modeling/ops.py

@@ -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)
 

tools/export_model.py

@@ -112,6 +112,8 @@ def dump_infer_config(FLAGS, config):
         'RetinaNet': 40,
         'Face': 3,
         'TTFNet': 3,
+        'FCOS': 3,
+        'EfficientDet': 40
     }
     infer_arch = config['architecture']