from transformers import PretrainedConfig
class VesselSegmentConfig(PretrainedConfig):
  model_type = "SegformerForSemanticSegmentation"
  def __init__(self, num_classes=1, input_channels=1, image_size=[512,512], features=[64,64,128], attention_dims=[64,32,16], patch_size=256, batch_size=4, **kwargs):
    self.num_classes = num_classes
    self.input_channels = input_channels
    self.image_size = image_size
    self.features = features
    self.attention_dims = attention_dims
    self.patch_size = patch_size
    self.batch_size = batch_size
    super().__init__(**kwargs)