dfine-medium / kf_config.json
IMvision12's picture
Upload kf_config.json with huggingface_hub
d629251 verified
Raw
History Blame Contribute Delete
1.15 kB
{
"library_name": "kerasformers",
"kerasformers_version": "1.1.3",
"model_module": "kerasformers.models.dfine",
"model_class": "DFineDetect",
"variant": "dfine-medium",
"weights": "model.weights.h5",
"model_type": "d_fine",
"stem_channels": [
3,
24,
32
],
"stage_in_channels": [
32,
96,
384,
768
],
"stage_mid_channels": [
32,
64,
128,
256
],
"stage_out_channels": [
96,
384,
768,
1536
],
"stage_num_blocks": [
1,
1,
3,
1
],
"stage_numb_of_layers": [
4,
4,
4,
4
],
"use_lab": true,
"encoder_in_channels": [
384,
768,
1536
],
"encoder_hidden_dim": 256,
"encoder_ffn_dim": 1024,
"encode_proj_layers": [
2
],
"hidden_expansion": 1.0,
"ccfm_num_blocks": 2,
"hidden_dim": 256,
"decoder_num_layers": 4,
"decoder_ffn_dim": 1024,
"decoder_n_points": [
3,
6,
3
],
"num_feature_levels": 3,
"feat_strides": [
8,
16,
32
],
"num_classes": 80,
"num_queries": 300,
"image_size": 640
}