Bryceee EasonFan commited on
Commit
decc736
·
verified ·
1 Parent(s): 6d3fdf4

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
+