CountGDPlusPlus / config.json
nikigoli's picture
Upload CountGD++ weights and code
383a563 verified
Raw History Blame Contribute Delete
3.55 kB
{
"add_channel_attention": false,
"add_pos_value": false,
"aux_loss": true,
"backbone": "swin_B_384_22k",
"backbone_freeze_keywords": null,
"batch_norm_type": "FrozenBatchNorm2d",
"batch_size": 4,
"bbox_loss_coef": 1.0,
"box_attn_type": "roi_align",
"box_threshold": 0.23,
"clip_max_norm": 0.1,
"cls_loss_coef": 5.0,
"dabdetr_deformable_decoder": false,
"dabdetr_deformable_encoder": false,
"dabdetr_yolo_like_anchor_update": false,
"data_aug_max_size": 1333,
"data_aug_scale_overlap": null,
"data_aug_scales": [
480,
512,
544,
576,
608,
640,
672,
704,
736,
768,
800
],
"data_aug_scales2_crop": [
384,
600
],
"data_aug_scales2_resize": [
400,
500,
600
],
"ddetr_lr_param": false,
"dec_layer_number": null,
"dec_layers": 6,
"dec_n_points": 4,
"dec_pred_bbox_embed_share": true,
"dec_pred_class_embed_share": true,
"decoder_layer_noise": false,
"decoder_module_seq": [
"sa",
"ca",
"ffn"
],
"decoder_sa_type": "sa",
"device": "cpu",
"dice_loss_coef": 1.0,
"dilation": false,
"dim_feedforward": 2048,
"dln_hw_noise": 0.2,
"dln_xy_noise": 0.2,
"dn_bbox_coef": 1.0,
"dn_box_noise_scale": 1.0,
"dn_label_coef": 1.0,
"dn_label_noise_ratio": 0.5,
"dn_labelbook_size": 91,
"dn_scalar": 100,
"dropout": 0.0,
"embed_init_tgt": true,
"enc_layers": 6,
"enc_loss_coef": 1.0,
"enc_n_points": 4,
"epochs": 30,
"fix_refpoints_hw": -1,
"focal_alpha": 0.25,
"focal_gamma": 2.0,
"freeze_keywords": [
"backbone.0",
"bert"
],
"frozen_weights": null,
"fusion_dropout": 0.0,
"fusion_droppath": 0.1,
"giou_loss_coef": 0.0,
"hidden_dim": 256,
"interm_loss_coef": 1.0,
"lr": 0.0001,
"lr_backbone": 1e-05,
"lr_backbone_names": [
"backbone.0",
"bert"
],
"lr_drop": 10,
"lr_drop_list": [
10,
20
],
"lr_linear_proj_mult": 1e-05,
"lr_linear_proj_names": [
"ref_point_head",
"sampling_offsets"
],
"mask_loss_coef": 1.0,
"masks": false,
"match_unstable_error": true,
"matcher_type": "HungarianMatcher",
"max_labels": 90,
"max_text_len": 256,
"modelname": "groundingdino",
"multi_step_lr": false,
"nheads": 8,
"nms_iou_threshold": -1,
"no_interm_box_loss": false,
"num_feature_levels": 4,
"num_patterns": 0,
"num_queries": 900,
"num_select": 900,
"onecyclelr": false,
"param_dict_type": "ddetr_in_mmdet",
"pdetr3_bbox_embed_diff_each_layer": false,
"pdetr3_refHW": -1,
"pe_temperatureH": 20,
"pe_temperatureW": 20,
"position_embedding": "sine",
"pre_norm": false,
"query_dim": 4,
"random_refpoints_xy": false,
"return_interm_indices": [
1,
2,
3
],
"save_checkpoint_interval": 10,
"set_cost_bbox": 1.0,
"set_cost_class": 5.0,
"set_cost_giou": 0.0,
"sub_sentence_present": true,
"text_dropout": 0.0,
"text_encoder_type": "bert-base-uncased",
"text_threshold": 0,
"transformer_activation": "relu",
"two_stage_add_query_num": 0,
"two_stage_bbox_embed_share": false,
"two_stage_class_embed_share": false,
"two_stage_default_hw": 0.05,
"two_stage_keep_all_tokens": false,
"two_stage_learn_wh": false,
"two_stage_pat_embed": 0,
"two_stage_type": "standard",
"use_checkpoint": false,
"use_coco_eval": false,
"use_deformable_box_attn": false,
"use_detached_boxes_dec_out": false,
"use_fusion_layer": true,
"use_text_cross_attention": true,
"use_text_enhancer": true,
"use_transformer_ckpt": false,
"weight_decay": 0.0001
}