[UPDATE] Replace edict with dict for configuration in anytext.py and RecModel.py for consistency

Author: tolgacangoz

examples/research_projects/anytext/anytext.py

@@ -348,11 +348,11 @@ def create_predictor(model_dir=None, model_lang="ch", device="cpu", use_fp16=Fal
         n_class = 97
     else:
         raise ValueError(f"Unsupported OCR recog model_lang: {model_lang}")
-    rec_config = edict(
+    rec_config = dict(
         in_channels=3,
-        backbone=edict(type="MobileNetV1Enhance", scale=0.5, last_conv_stride=[1, 2], last_pool_type="avg"),
-        neck=edict(type="SequenceEncoder", encoder_type="svtr", dims=64, depth=2, hidden_dims=120, use_guide=True),
-        head=edict(type="CTCHead", fc_decay=0.00001, out_channels=n_class, return_feats=True),
+        backbone=dict(type="MobileNetV1Enhance", scale=0.5, last_conv_stride=[1, 2], last_pool_type="avg"),
+        neck=dict(type="SequenceEncoder", encoder_type="svtr", dims=64, depth=2, hidden_dims=120, use_guide=True),
+        head=dict(type="CTCHead", fc_decay=0.00001, out_channels=n_class, return_feats=True),
     )
 
     rec_model = RecModel(rec_config)

examples/research_projects/anytext/ocr_recog/RecModel.py

@@ -14,17 +14,17 @@ class RecModel(nn.Module):
     def __init__(self, config):
         super().__init__()
         assert "in_channels" in config, "in_channels must in model config"
-        backbone_type = config.backbone.pop("type")
+        backbone_type = config["backbone"].pop("type")
         assert backbone_type in backbone_dict, f"backbone.type must in {backbone_dict}"
-        self.backbone = backbone_dict[backbone_type](config.in_channels, **config.backbone)
+        self.backbone = backbone_dict[backbone_type](config['in_channels'], **config['backbone'])
 
-        neck_type = config.neck.pop("type")
+        neck_type = config['neck'].pop("type")
         assert neck_type in neck_dict, f"neck.type must in {neck_dict}"
-        self.neck = neck_dict[neck_type](self.backbone.out_channels, **config.neck)
+        self.neck = neck_dict[neck_type](self.backbone.out_channels, **config['neck'])
 
-        head_type = config.head.pop("type")
+        head_type = config['head'].pop("type")
         assert head_type in head_dict, f"head.type must in {head_dict}"
-        self.head = head_dict[head_type](self.neck.out_channels, **config.head)
+        self.head = head_dict[head_type](self.neck.out_channels, **config['head'])
 
         self.name = f"RecModel_{backbone_type}_{neck_type}_{head_type}"