config.json

{
    "module": "keras_hub.src.models.d_fine.d_fine_backbone",
    "class_name": "DFineBackbone",
    "config": {
        "name": "d_fine_backbone",
        "trainable": true,
        "backbone": {
            "module": "keras_hub.src.models.hgnetv2.hgnetv2_backbone",
            "class_name": "HGNetV2Backbone",
            "config": {
                "name": "hg_net_v2_backbone",
                "trainable": true,
                "depths": [
                    3,
                    4,
                    6,
                    3
                ],
                "embedding_size": 32,
                "hidden_sizes": [
                    192,
                    384,
                    768,
                    1536
                ],
                "stem_channels": [
                    3,
                    24,
                    32
                ],
                "hidden_act": "relu",
                "use_learnable_affine_block": true,
                "stackwise_stage_filters": [
                    [
                        32,
                        32,
                        96,
                        1,
                        4,
                        3
                    ],
                    [
                        96,
                        64,
                        384,
                        1,
                        4,
                        3
                    ],
                    [
                        384,
                        128,
                        768,
                        3,
                        4,
                        5
                    ],
                    [
                        768,
                        256,
                        1536,
                        1,
                        4,
                        5
                    ]
                ],
                "apply_downsample": [
                    false,
                    true,
                    true,
                    true
                ],
                "use_lightweight_conv_block": [
                    false,
                    false,
                    true,
                    true
                ],
                "image_shape": [
                    null,
                    null,
                    3
                ],
                "out_features": [
                    "stage2",
                    "stage3",
                    "stage4"
                ],
                "data_format": "channels_last"
            },
            "registered_name": "keras_hub>HGNetV2Backbone"
        },
        "decoder_in_channels": [
            256,
            256,
            256
        ],
        "encoder_hidden_dim": 256,
        "num_labels": 366,
        "num_denoising": 100,
        "learn_initial_query": false,
        "num_queries": 300,
        "anchor_image_size": [
            640,
            640
        ],
        "feat_strides": [
            8,
            16,
            32
        ],
        "num_feature_levels": 3,
        "hidden_dim": 256,
        "encoder_in_channels": [
            384,
            768,
            1536
        ],
        "encode_proj_layers": [
            2
        ],
        "num_attention_heads": 8,
        "encoder_ffn_dim": 1024,
        "num_encoder_layers": 1,
        "hidden_expansion": 1.0,
        "depth_multiplier": 0.67,
        "eval_idx": -1,
        "box_noise_scale": 1.0,
        "label_noise_ratio": 0.5,
        "labels": null,
        "num_decoder_layers": 4,
        "decoder_attention_heads": 8,
        "decoder_ffn_dim": 1024,
        "decoder_method": "default",
        "decoder_n_points": [
            3,
            6,
            3
        ],
        "lqe_hidden_dim": 64,
        "num_lqe_layers": 2,
        "seed": 0,
        "image_shape": [
            null,
            null,
            3
        ],
        "data_format": "channels_last",
        "out_features": [
            "stage2",
            "stage3",
            "stage4"
        ]
    },
    "registered_name": "keras_hub>DFineBackbone"
}