Fix missing config attributes and imports

app.py

@@ -56,6 +56,7 @@ class Config:
         # Visual backbone
         self.visual_backbone = "ResNet10"
         self.diagram_size = 128
+        self.pretrain_vis_path = ''  # Added missing attribute
         
         # Encoder
         self.encoder_type = "gru"
@@ -80,6 +81,9 @@ class Config:
         
         # Dataset
         self.without_stru = False
+        
+        # Logger (dummy for compatibility)
+        self.logger = type('obj', (object,), {'info': lambda x: print(x)})
 
 # Initialize model
 def load_model():

model/backbone/__init__.py

@@ -1,16 +1,18 @@
 from .resnet import *
 from .mobilenet_v2 import *
-from config import visual_backbone_list
 
+visual_backbone_list = ['ResNet10', 'mobilenet_v2']
 
 def get_visual_backbone(args):
     if args.visual_backbone in visual_backbone_list:
         model = eval(args.visual_backbone)()
-        if args.pretrain_vis_path !="":
+        if hasattr(args, 'pretrain_vis_path') and args.pretrain_vis_path != "":
             model.load_model(pretrain=args.pretrain_vis_path)
-            args.logger.info("Visual backbone has been loaded...")
+            if hasattr(args, 'logger'):
+                args.logger.info("Visual backbone has been loaded...")
         else:
-            args.logger.info("Visual backbone choose to train from scratch")
+            if hasattr(args, 'logger'):
+                args.logger.info("Visual backbone choose to train from scratch")
         return model
     else:
         raise NotImplementedError("Unsupported Backbone: {}".format(args.visual_backbone))

model/decoder/__init__.py

@@ -1,9 +1,10 @@
 # from .transformer import TransformerModel
-from config import decoder_list
 from .rnn_decoder import DecoderRNN
 from .tree_decoder import TreeDecoder
 from .transformer import TransformerDecoder
 
+decoder_list = ["rnn_decoder", "tree_decoder", "transformer"]
+
 def get_decoder(params, *args):
          
     if not params.decoder_type in decoder_list:

model/encoder/__init__.py

@@ -1,8 +1,9 @@
 from .lstm import LSTM
 from .gru import GRU
-from config import encoder_list
 from .transformer import TransformerEncoder
 
+encoder_list = ['lstm', 'gru', 'transformer']
+
 def get_encoder(params, *args):
 
     if not params.encoder_type in encoder_list: