Upload 3 files (#2)
Browse files- Upload 3 files (4332c574862b48a44ac305279517075485c472ef)
Co-authored-by: fan <[email protected]>
Teeth_Visual_Experts_DINO_5scale_Swinl_paranomic_x-ray_caries_filling.py
ADDED
|
@@ -0,0 +1,114 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
_base_ = ['coco_transformer.py']
|
| 2 |
+
|
| 3 |
+
num_classes=3
|
| 4 |
+
|
| 5 |
+
lr = 0.0001
|
| 6 |
+
param_dict_type = 'default'
|
| 7 |
+
lr_backbone = 1e-05
|
| 8 |
+
lr_backbone_names = ['backbone.0']
|
| 9 |
+
lr_linear_proj_names = ['reference_points', 'sampling_offsets']
|
| 10 |
+
lr_linear_proj_mult = 0.1
|
| 11 |
+
ddetr_lr_param = False
|
| 12 |
+
batch_size = 2
|
| 13 |
+
weight_decay = 0.0001
|
| 14 |
+
epochs = 36
|
| 15 |
+
lr_drop = 11
|
| 16 |
+
save_checkpoint_interval = 1
|
| 17 |
+
clip_max_norm = 0.1
|
| 18 |
+
onecyclelr = False
|
| 19 |
+
multi_step_lr = False
|
| 20 |
+
lr_drop_list = [33, 45]
|
| 21 |
+
|
| 22 |
+
|
| 23 |
+
modelname = 'dino'
|
| 24 |
+
frozen_weights = None
|
| 25 |
+
backbone = 'swin_L_384_22k'
|
| 26 |
+
use_checkpoint = True
|
| 27 |
+
|
| 28 |
+
dilation = False
|
| 29 |
+
position_embedding = 'sine'
|
| 30 |
+
pe_temperatureH = 20
|
| 31 |
+
pe_temperatureW = 20
|
| 32 |
+
# 修改特征层级索引,增加0层
|
| 33 |
+
return_interm_indices = [0, 1, 2, 3] # 改为4个层级,从0开始
|
| 34 |
+
backbone_freeze_keywords = None
|
| 35 |
+
enc_layers = 6
|
| 36 |
+
dec_layers = 6
|
| 37 |
+
unic_layers = 0
|
| 38 |
+
pre_norm = False
|
| 39 |
+
dim_feedforward = 2048
|
| 40 |
+
hidden_dim = 256
|
| 41 |
+
dropout = 0.0
|
| 42 |
+
nheads = 8
|
| 43 |
+
num_queries = 900
|
| 44 |
+
query_dim = 4
|
| 45 |
+
num_patterns = 0
|
| 46 |
+
pdetr3_bbox_embed_diff_each_layer = False
|
| 47 |
+
pdetr3_refHW = -1
|
| 48 |
+
random_refpoints_xy = False
|
| 49 |
+
fix_refpoints_hw = -1
|
| 50 |
+
dabdetr_yolo_like_anchor_update = False
|
| 51 |
+
dabdetr_deformable_encoder = False
|
| 52 |
+
dabdetr_deformable_decoder = False
|
| 53 |
+
use_deformable_box_attn = False
|
| 54 |
+
box_attn_type = 'roi_align'
|
| 55 |
+
dec_layer_number = None
|
| 56 |
+
num_feature_levels = 5 # 改为5个特征层级
|
| 57 |
+
enc_n_points = 4
|
| 58 |
+
dec_n_points = 4
|
| 59 |
+
decoder_layer_noise = False
|
| 60 |
+
dln_xy_noise = 0.2
|
| 61 |
+
dln_hw_noise = 0.2
|
| 62 |
+
add_channel_attention = False
|
| 63 |
+
add_pos_value = False
|
| 64 |
+
two_stage_type = 'standard'
|
| 65 |
+
two_stage_pat_embed = 0
|
| 66 |
+
two_stage_add_query_num = 0
|
| 67 |
+
two_stage_bbox_embed_share = False
|
| 68 |
+
two_stage_class_embed_share = False
|
| 69 |
+
two_stage_learn_wh = False
|
| 70 |
+
two_stage_default_hw = 0.05
|
| 71 |
+
two_stage_keep_all_tokens = False
|
| 72 |
+
num_select = 900
|
| 73 |
+
transformer_activation = 'relu'
|
| 74 |
+
batch_norm_type = 'FrozenBatchNorm2d'
|
| 75 |
+
masks = False
|
| 76 |
+
aux_loss = True
|
| 77 |
+
set_cost_class = 2.0
|
| 78 |
+
set_cost_bbox = 5.0
|
| 79 |
+
set_cost_giou = 2.0
|
| 80 |
+
cls_loss_coef = 1.0
|
| 81 |
+
mask_loss_coef = 1.0
|
| 82 |
+
dice_loss_coef = 1.0
|
| 83 |
+
bbox_loss_coef = 5.0
|
| 84 |
+
giou_loss_coef = 2.0
|
| 85 |
+
enc_loss_coef = 1.0
|
| 86 |
+
interm_loss_coef = 1.0
|
| 87 |
+
no_interm_box_loss = False
|
| 88 |
+
focal_alpha = 0.25
|
| 89 |
+
|
| 90 |
+
decoder_sa_type = 'sa' # ['sa', 'ca_label', 'ca_content']
|
| 91 |
+
matcher_type = 'HungarianMatcher' # or SimpleMinsumMatcher
|
| 92 |
+
decoder_module_seq = ['sa', 'ca', 'ffn']
|
| 93 |
+
nms_iou_threshold = -1
|
| 94 |
+
|
| 95 |
+
dec_pred_bbox_embed_share = True
|
| 96 |
+
dec_pred_class_embed_share = True
|
| 97 |
+
|
| 98 |
+
# for dn
|
| 99 |
+
use_dn = True
|
| 100 |
+
dn_number = 100
|
| 101 |
+
dn_box_noise_scale = 0.4
|
| 102 |
+
dn_label_noise_ratio = 0.5
|
| 103 |
+
embed_init_tgt = True
|
| 104 |
+
dn_labelbook_size = 3
|
| 105 |
+
|
| 106 |
+
match_unstable_error = True
|
| 107 |
+
|
| 108 |
+
# for ema
|
| 109 |
+
use_ema = False
|
| 110 |
+
ema_decay = 0.9997
|
| 111 |
+
ema_epoch = 0
|
| 112 |
+
|
| 113 |
+
use_detached_boxes_dec_out = False
|
| 114 |
+
|
Teeth_Visual_Experts_DINO_Swinl_5scale_panoramic_x-ray_4diseases.py
ADDED
|
@@ -0,0 +1,114 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
_base_ = ['coco_transformer.py']
|
| 2 |
+
|
| 3 |
+
num_classes=5
|
| 4 |
+
|
| 5 |
+
lr = 0.0001
|
| 6 |
+
param_dict_type = 'default'
|
| 7 |
+
lr_backbone = 1e-05
|
| 8 |
+
lr_backbone_names = ['backbone.0']
|
| 9 |
+
lr_linear_proj_names = ['reference_points', 'sampling_offsets']
|
| 10 |
+
lr_linear_proj_mult = 0.1
|
| 11 |
+
ddetr_lr_param = False
|
| 12 |
+
batch_size = 2
|
| 13 |
+
weight_decay = 0.0001
|
| 14 |
+
epochs = 36
|
| 15 |
+
lr_drop = 11
|
| 16 |
+
save_checkpoint_interval = 1
|
| 17 |
+
clip_max_norm = 0.1
|
| 18 |
+
onecyclelr = False
|
| 19 |
+
multi_step_lr = False
|
| 20 |
+
lr_drop_list = [33, 45]
|
| 21 |
+
|
| 22 |
+
|
| 23 |
+
modelname = 'dino'
|
| 24 |
+
frozen_weights = None
|
| 25 |
+
backbone = 'swin_L_384_22k'
|
| 26 |
+
use_checkpoint = True
|
| 27 |
+
|
| 28 |
+
dilation = False
|
| 29 |
+
position_embedding = 'sine'
|
| 30 |
+
pe_temperatureH = 20
|
| 31 |
+
pe_temperatureW = 20
|
| 32 |
+
# 修改特征层级索引,增加0层
|
| 33 |
+
return_interm_indices = [0, 1, 2, 3] # 改为4个层级,从0开始
|
| 34 |
+
backbone_freeze_keywords = None
|
| 35 |
+
enc_layers = 6
|
| 36 |
+
dec_layers = 6
|
| 37 |
+
unic_layers = 0
|
| 38 |
+
pre_norm = False
|
| 39 |
+
dim_feedforward = 2048
|
| 40 |
+
hidden_dim = 256
|
| 41 |
+
dropout = 0.0
|
| 42 |
+
nheads = 8
|
| 43 |
+
num_queries = 900
|
| 44 |
+
query_dim = 4
|
| 45 |
+
num_patterns = 0
|
| 46 |
+
pdetr3_bbox_embed_diff_each_layer = False
|
| 47 |
+
pdetr3_refHW = -1
|
| 48 |
+
random_refpoints_xy = False
|
| 49 |
+
fix_refpoints_hw = -1
|
| 50 |
+
dabdetr_yolo_like_anchor_update = False
|
| 51 |
+
dabdetr_deformable_encoder = False
|
| 52 |
+
dabdetr_deformable_decoder = False
|
| 53 |
+
use_deformable_box_attn = False
|
| 54 |
+
box_attn_type = 'roi_align'
|
| 55 |
+
dec_layer_number = None
|
| 56 |
+
num_feature_levels = 5 # 改为5个特征层级
|
| 57 |
+
enc_n_points = 4
|
| 58 |
+
dec_n_points = 4
|
| 59 |
+
decoder_layer_noise = False
|
| 60 |
+
dln_xy_noise = 0.2
|
| 61 |
+
dln_hw_noise = 0.2
|
| 62 |
+
add_channel_attention = False
|
| 63 |
+
add_pos_value = False
|
| 64 |
+
two_stage_type = 'standard'
|
| 65 |
+
two_stage_pat_embed = 0
|
| 66 |
+
two_stage_add_query_num = 0
|
| 67 |
+
two_stage_bbox_embed_share = False
|
| 68 |
+
two_stage_class_embed_share = False
|
| 69 |
+
two_stage_learn_wh = False
|
| 70 |
+
two_stage_default_hw = 0.05
|
| 71 |
+
two_stage_keep_all_tokens = False
|
| 72 |
+
num_select = 900
|
| 73 |
+
transformer_activation = 'relu'
|
| 74 |
+
batch_norm_type = 'FrozenBatchNorm2d'
|
| 75 |
+
masks = False
|
| 76 |
+
aux_loss = True
|
| 77 |
+
set_cost_class = 2.0
|
| 78 |
+
set_cost_bbox = 5.0
|
| 79 |
+
set_cost_giou = 2.0
|
| 80 |
+
cls_loss_coef = 1.0
|
| 81 |
+
mask_loss_coef = 1.0
|
| 82 |
+
dice_loss_coef = 1.0
|
| 83 |
+
bbox_loss_coef = 5.0
|
| 84 |
+
giou_loss_coef = 2.0
|
| 85 |
+
enc_loss_coef = 1.0
|
| 86 |
+
interm_loss_coef = 1.0
|
| 87 |
+
no_interm_box_loss = False
|
| 88 |
+
focal_alpha = 0.25
|
| 89 |
+
|
| 90 |
+
decoder_sa_type = 'sa' # ['sa', 'ca_label', 'ca_content']
|
| 91 |
+
matcher_type = 'HungarianMatcher' # or SimpleMinsumMatcher
|
| 92 |
+
decoder_module_seq = ['sa', 'ca', 'ffn']
|
| 93 |
+
nms_iou_threshold = -1
|
| 94 |
+
|
| 95 |
+
dec_pred_bbox_embed_share = True
|
| 96 |
+
dec_pred_class_embed_share = True
|
| 97 |
+
|
| 98 |
+
# for dn
|
| 99 |
+
use_dn = True
|
| 100 |
+
dn_number = 100
|
| 101 |
+
dn_box_noise_scale = 0.4
|
| 102 |
+
dn_label_noise_ratio = 0.5
|
| 103 |
+
embed_init_tgt = True
|
| 104 |
+
dn_labelbook_size = 5
|
| 105 |
+
|
| 106 |
+
match_unstable_error = True
|
| 107 |
+
|
| 108 |
+
# for ema
|
| 109 |
+
use_ema = False
|
| 110 |
+
ema_decay = 0.9997
|
| 111 |
+
ema_epoch = 0
|
| 112 |
+
|
| 113 |
+
use_detached_boxes_dec_out = False
|
| 114 |
+
|
Teeth_Visual_Experts_DINO_Swinl_5scale_paranomic_x-ray_12diseases.py
ADDED
|
@@ -0,0 +1,114 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
_base_ = ['coco_transformer.py']
|
| 2 |
+
|
| 3 |
+
num_classes=12
|
| 4 |
+
|
| 5 |
+
lr = 0.0001
|
| 6 |
+
param_dict_type = 'default'
|
| 7 |
+
lr_backbone = 1e-05
|
| 8 |
+
lr_backbone_names = ['backbone.0']
|
| 9 |
+
lr_linear_proj_names = ['reference_points', 'sampling_offsets']
|
| 10 |
+
lr_linear_proj_mult = 0.1
|
| 11 |
+
ddetr_lr_param = False
|
| 12 |
+
batch_size = 2
|
| 13 |
+
weight_decay = 0.0001
|
| 14 |
+
epochs = 36
|
| 15 |
+
lr_drop = 11
|
| 16 |
+
save_checkpoint_interval = 1
|
| 17 |
+
clip_max_norm = 0.1
|
| 18 |
+
onecyclelr = False
|
| 19 |
+
multi_step_lr = False
|
| 20 |
+
lr_drop_list = [33, 45]
|
| 21 |
+
|
| 22 |
+
|
| 23 |
+
modelname = 'dino'
|
| 24 |
+
frozen_weights = None
|
| 25 |
+
backbone = 'swin_L_384_22k'
|
| 26 |
+
use_checkpoint = True
|
| 27 |
+
|
| 28 |
+
dilation = False
|
| 29 |
+
position_embedding = 'sine'
|
| 30 |
+
pe_temperatureH = 20
|
| 31 |
+
pe_temperatureW = 20
|
| 32 |
+
# 修改特征层级索引,增加0层
|
| 33 |
+
return_interm_indices = [0, 1, 2, 3] # 改为4个层级,从0开始
|
| 34 |
+
backbone_freeze_keywords = None
|
| 35 |
+
enc_layers = 6
|
| 36 |
+
dec_layers = 6
|
| 37 |
+
unic_layers = 0
|
| 38 |
+
pre_norm = False
|
| 39 |
+
dim_feedforward = 2048
|
| 40 |
+
hidden_dim = 256
|
| 41 |
+
dropout = 0.0
|
| 42 |
+
nheads = 8
|
| 43 |
+
num_queries = 900
|
| 44 |
+
query_dim = 4
|
| 45 |
+
num_patterns = 0
|
| 46 |
+
pdetr3_bbox_embed_diff_each_layer = False
|
| 47 |
+
pdetr3_refHW = -1
|
| 48 |
+
random_refpoints_xy = False
|
| 49 |
+
fix_refpoints_hw = -1
|
| 50 |
+
dabdetr_yolo_like_anchor_update = False
|
| 51 |
+
dabdetr_deformable_encoder = False
|
| 52 |
+
dabdetr_deformable_decoder = False
|
| 53 |
+
use_deformable_box_attn = False
|
| 54 |
+
box_attn_type = 'roi_align'
|
| 55 |
+
dec_layer_number = None
|
| 56 |
+
num_feature_levels = 5 # 改为5个特征层级
|
| 57 |
+
enc_n_points = 4
|
| 58 |
+
dec_n_points = 4
|
| 59 |
+
decoder_layer_noise = False
|
| 60 |
+
dln_xy_noise = 0.2
|
| 61 |
+
dln_hw_noise = 0.2
|
| 62 |
+
add_channel_attention = False
|
| 63 |
+
add_pos_value = False
|
| 64 |
+
two_stage_type = 'standard'
|
| 65 |
+
two_stage_pat_embed = 0
|
| 66 |
+
two_stage_add_query_num = 0
|
| 67 |
+
two_stage_bbox_embed_share = False
|
| 68 |
+
two_stage_class_embed_share = False
|
| 69 |
+
two_stage_learn_wh = False
|
| 70 |
+
two_stage_default_hw = 0.05
|
| 71 |
+
two_stage_keep_all_tokens = False
|
| 72 |
+
num_select = 900
|
| 73 |
+
transformer_activation = 'relu'
|
| 74 |
+
batch_norm_type = 'FrozenBatchNorm2d'
|
| 75 |
+
masks = False
|
| 76 |
+
aux_loss = True
|
| 77 |
+
set_cost_class = 2.0
|
| 78 |
+
set_cost_bbox = 5.0
|
| 79 |
+
set_cost_giou = 2.0
|
| 80 |
+
cls_loss_coef = 1.0
|
| 81 |
+
mask_loss_coef = 1.0
|
| 82 |
+
dice_loss_coef = 1.0
|
| 83 |
+
bbox_loss_coef = 5.0
|
| 84 |
+
giou_loss_coef = 2.0
|
| 85 |
+
enc_loss_coef = 1.0
|
| 86 |
+
interm_loss_coef = 1.0
|
| 87 |
+
no_interm_box_loss = False
|
| 88 |
+
focal_alpha = 0.25
|
| 89 |
+
|
| 90 |
+
decoder_sa_type = 'sa' # ['sa', 'ca_label', 'ca_content']
|
| 91 |
+
matcher_type = 'HungarianMatcher' # or SimpleMinsumMatcher
|
| 92 |
+
decoder_module_seq = ['sa', 'ca', 'ffn']
|
| 93 |
+
nms_iou_threshold = -1
|
| 94 |
+
|
| 95 |
+
dec_pred_bbox_embed_share = True
|
| 96 |
+
dec_pred_class_embed_share = True
|
| 97 |
+
|
| 98 |
+
# for dn
|
| 99 |
+
use_dn = True
|
| 100 |
+
dn_number = 100
|
| 101 |
+
dn_box_noise_scale = 0.4
|
| 102 |
+
dn_label_noise_ratio = 0.5
|
| 103 |
+
embed_init_tgt = True
|
| 104 |
+
dn_labelbook_size = 13
|
| 105 |
+
|
| 106 |
+
match_unstable_error = True
|
| 107 |
+
|
| 108 |
+
# for ema
|
| 109 |
+
use_ema = False
|
| 110 |
+
ema_decay = 0.9997
|
| 111 |
+
ema_epoch = 0
|
| 112 |
+
|
| 113 |
+
use_detached_boxes_dec_out = False
|
| 114 |
+
|