config.json

{
    "module": "keras_hub.src.models.segformer.segformer_backbone",
    "class_name": "SegFormerBackbone",
    "config": {
        "name": "seg_former_backbone",
        "trainable": true,
        "projection_filters": 256,
        "image_encoder": {
            "module": "keras_hub.src.models.mit.mit_backbone",
            "class_name": "MiTBackbone",
            "config": {
                "name": "mi_t_backbone",
                "trainable": true,
                "layerwise_depths": [
                    2,
                    2,
                    2,
                    2
                ],
                "hidden_dims": [
                    32,
                    64,
                    160,
                    256
                ],
                "image_shape": [
                    512,
                    512,
                    3
                ],
                "num_layers": 4,
                "layerwise_num_heads": [
                    1,
                    2,
                    5,
                    8
                ],
                "layerwise_sr_ratios": [
                    8,
                    4,
                    2,
                    1
                ],
                "max_drop_path_rate": 0.1,
                "layerwise_patch_sizes": [
                    7,
                    3,
                    3,
                    3
                ],
                "layerwise_strides": [
                    4,
                    2,
                    2,
                    2
                ]
            },
            "registered_name": "keras_hub>MiTBackbone"
        }
    },
    "registered_name": "keras_hub>SegFormerBackbone"
}