| | _base_ = ['coco_transformer.py'] |
| |
|
| | num_classes=33 |
| |
|
| | lr = 0.0001 |
| | param_dict_type = 'default' |
| | lr_backbone = 1e-05 |
| | lr_backbone_names = ['backbone.0'] |
| | lr_linear_proj_names = ['reference_points', 'sampling_offsets'] |
| | lr_linear_proj_mult = 0.1 |
| | ddetr_lr_param = False |
| | batch_size = 4 |
| | weight_decay = 0.0001 |
| | epochs = 50 |
| | lr_drop = 40 |
| | save_checkpoint_interval = 1 |
| | clip_max_norm = 0.1 |
| | onecyclelr = False |
| | multi_step_lr = False |
| | lr_drop_list = [33, 45] |
| |
|
| |
|
| | modelname = 'dino' |
| | frozen_weights = None |
| | backbone = 'swin_L_384_22k' |
| | use_checkpoint = True |
| |
|
| | dilation = False |
| | position_embedding = 'sine' |
| | pe_temperatureH = 20 |
| | pe_temperatureW = 20 |
| | return_interm_indices = [1, 2, 3] |
| | backbone_freeze_keywords = None |
| | enc_layers = 6 |
| | dec_layers = 6 |
| | unic_layers = 0 |
| | pre_norm = False |
| | dim_feedforward = 2048 |
| | hidden_dim = 256 |
| | dropout = 0.0 |
| | nheads = 8 |
| | num_queries = 900 |
| | query_dim = 4 |
| | num_patterns = 0 |
| | pdetr3_bbox_embed_diff_each_layer = False |
| | pdetr3_refHW = -1 |
| | random_refpoints_xy = False |
| | fix_refpoints_hw = -1 |
| | dabdetr_yolo_like_anchor_update = False |
| | dabdetr_deformable_encoder = False |
| | dabdetr_deformable_decoder = False |
| | use_deformable_box_attn = False |
| | box_attn_type = 'roi_align' |
| | dec_layer_number = None |
| | num_feature_levels = 5 |
| | enc_n_points = 4 |
| | dec_n_points = 4 |
| | decoder_layer_noise = False |
| | dln_xy_noise = 0.2 |
| | dln_hw_noise = 0.2 |
| | add_channel_attention = False |
| | add_pos_value = False |
| | two_stage_type = 'standard' |
| | two_stage_pat_embed = 0 |
| | two_stage_add_query_num = 0 |
| | two_stage_bbox_embed_share = False |
| | two_stage_class_embed_share = False |
| | two_stage_learn_wh = False |
| | two_stage_default_hw = 0.05 |
| | two_stage_keep_all_tokens = False |
| | num_select = 300 |
| | transformer_activation = 'relu' |
| | batch_norm_type = 'FrozenBatchNorm2d' |
| | masks = False |
| | aux_loss = True |
| | set_cost_class = 2.0 |
| | set_cost_bbox = 5.0 |
| | set_cost_giou = 2.0 |
| | cls_loss_coef = 1.0 |
| | mask_loss_coef = 1.0 |
| | dice_loss_coef = 1.0 |
| | bbox_loss_coef = 5.0 |
| | giou_loss_coef = 2.0 |
| | enc_loss_coef = 1.0 |
| | interm_loss_coef = 1.0 |
| | no_interm_box_loss = False |
| | focal_alpha = 0.25 |
| |
|
| | decoder_sa_type = 'sa' |
| | matcher_type = 'HungarianMatcher' |
| | decoder_module_seq = ['sa', 'ca', 'ffn'] |
| | nms_iou_threshold = -1 |
| |
|
| | dec_pred_bbox_embed_share = True |
| | dec_pred_class_embed_share = True |
| |
|
| | |
| | use_dn = True |
| | dn_number = 100 |
| | dn_box_noise_scale = 0.4 |
| | dn_label_noise_ratio = 0.5 |
| | embed_init_tgt = True |
| | dn_labelbook_size = 33 |
| |
|
| | match_unstable_error = True |
| |
|
| | |
| | use_ema = False |
| | ema_decay = 0.9997 |
| | ema_epoch = 0 |
| |
|
| | use_detached_boxes_dec_out = False |
| |
|
| |
|
