optimize code style

Author: yjh0410

.gitignore

@@ -8,7 +8,3 @@ weights
 __pycache__
 det_results
 .vscode
-deployment/OpenVINO/cpp/build
-train_nebula.py
-train_nebula.sh
-cluster.json

config/__init__.py

@@ -1,21 +1,22 @@
 # ----------------------- Model Config -----------------------
-from .retinanet_config  import retinanet_cfg
-from .fcos_config       import fcos_cfg
-from .yolof_config      import yolof_cfg
-from .plain_detr_config import plain_detr_cfg
+from .fcos_config      import build_fcos_config
+from .yolof_config     import build_yolof_config
+from .detr_config      import build_detr_config
 
 def build_config(args):
-    # RetinaNet
-    if args.model in retinanet_cfg.keys():
-        return retinanet_cfg[args.model]
     # FCOS
-    elif args.model in fcos_cfg.keys():
-        return fcos_cfg[args.model]
+    if "fcos" in args.model:
+        cfg = build_fcos_config(args)
     # YOLOF
-    elif args.model in yolof_cfg.keys():
-        return yolof_cfg[args.model]
-    # PlainDETR
-    elif args.model in plain_detr_cfg.keys():
-        return plain_detr_cfg[args.model]
+    elif "yolof" in args.model:
+        cfg = build_yolof_config(args)
+    # DETR
+    elif "detr" in args.model:
+        cfg = build_detr_config(args)
     else:
         raise NotImplementedError('Unknown Model: {}'.format(args.model))
+
+    # Print model config
+    cfg.print_config()
+
+    return cfg

config/detr_config.py

@@ -0,0 +1,23 @@
+# End-to-end Detection with Transformer
+
+def build_detr_config(args):
+    if   args.model == 'detr_r50':
+        return Detr_R50_Config()
+    else:
+        raise NotImplementedError("No config for model: {}".format(args.model))
+
+
+class DetrBaseConfig(object):
+    def __init__(self):
+        pass
+
+    def print_config(self):
+        config_dict = {key: value for key, value in self.__dict__.items() if not key.startswith('__')}
+        for k, v in config_dict.items():
+            print("{} : {}".format(k, v))
+
+class Detr_R50_Config(DetrBaseConfig):
+    def __init__(self) -> None:
+        super().__init__()
+        ## Backbone
+        pass