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