|
|
_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 |
|
|
|
|
|
|