rtmw-l-384x288 / config.json
akore's picture
Add rtmw-l-384x288 RTMW/RTMDet HF port
5bb04af verified
{
"backbone_arch": "P5",
"backbone_channel_attention": true,
"backbone_deepen_factor": 1.0,
"backbone_expand_ratio": 0.5,
"backbone_widen_factor": 1.0,
"decoder_normalize": false,
"decoder_sigma": [
6.0,
6.93
],
"decoder_use_dark": false,
"gau_act_fn": "SiLU",
"gau_drop_path": 0.0,
"gau_dropout_rate": 0.0,
"gau_expansion_factor": 2,
"gau_hidden_dims": 256,
"gau_pos_enc": false,
"gau_s": 128,
"gau_use_rel_bias": false,
"head_final_layer_kernel_size": 7,
"head_in_channels": 1024,
"head_in_featuremap_size": [
9,
12
],
"input_size": [
288,
384
],
"model_type": "rtmw",
"neck_expand_ratio": 0.5,
"neck_in_channels": [
256,
512,
1024
],
"neck_num_csp_blocks": 2,
"neck_out_channels": null,
"num_keypoints": 133,
"score_max": 8.0,
"score_min": 0.0,
"simcc_split_ratio": 2.0,
"transformers_version": "5.2.0",
"auto_map": {
"AutoConfig": "configuration_rtmw.RTMWConfig",
"AutoModel": "modeling_rtmw.RTMWModel"
}
}