File size: 3,547 Bytes
383a563 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 | {
"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
} |