|
|
@@ -226,6 +226,62 @@ rtcdet_cfg = {
|
|
|
'trainer_type': 'rtcdet',
|
|
|
},
|
|
|
|
|
|
+ 'rtcdet_m':{
|
|
|
+ # ---------------- Model config ----------------
|
|
|
+ ## Backbone
|
|
|
+ 'backbone': 'elannet_v2',
|
|
|
+ 'pretrained': False,
|
|
|
+ 'bk_act': 'silu',
|
|
|
+ 'bk_norm': 'BN',
|
|
|
+ 'bk_depthwise': False,
|
|
|
+ 'width': 0.75,
|
|
|
+ 'depth': 0.67,
|
|
|
+ 'stride': [8, 16, 32], # P3, P4, P5
|
|
|
+ 'max_stride': 32,
|
|
|
+ 'reg_max': 16,
|
|
|
+ ## Neck: SPP
|
|
|
+ 'neck': 'sppf',
|
|
|
+ 'neck_expand_ratio': 0.5,
|
|
|
+ 'pooling_size': 5,
|
|
|
+ 'neck_act': 'silu',
|
|
|
+ 'neck_norm': 'BN',
|
|
|
+ 'neck_depthwise': False,
|
|
|
+ ## Neck: PaFPN
|
|
|
+ 'fpn': 'rtcdet_pafpn',
|
|
|
+ 'fpn_reduce_layer': 'conv',
|
|
|
+ 'fpn_downsample_layer': 'dsblock',
|
|
|
+ 'fpn_core_block': 'elan_block',
|
|
|
+ 'fpn_branch_depth': 3,
|
|
|
+ 'fpn_expand_ratio': 0.5,
|
|
|
+ 'fpn_act': 'silu',
|
|
|
+ 'fpn_norm': 'BN',
|
|
|
+ 'fpn_depthwise': False,
|
|
|
+ ## Head
|
|
|
+ 'head': 'decoupled_head',
|
|
|
+ 'head_act': 'silu',
|
|
|
+ 'head_norm': 'BN',
|
|
|
+ 'num_cls_head': 2,
|
|
|
+ 'num_reg_head': 2,
|
|
|
+ 'head_depthwise': False,
|
|
|
+ # ---------------- Train config ----------------
|
|
|
+ ## Input
|
|
|
+ 'multi_scale': [0.5, 1.25], # 320 -> 800
|
|
|
+ 'trans_type': 'yolox_medium',
|
|
|
+ # ---------------- Assignment config ----------------
|
|
|
+ ## Matcher
|
|
|
+ 'matcher': {'center_sampling_radius': 2.5,
|
|
|
+ 'topk_candidate': 10},
|
|
|
+ # ---------------- Loss config ----------------
|
|
|
+ ## Loss weight
|
|
|
+ 'ema_update': False,
|
|
|
+ 'loss_box_aux': True,
|
|
|
+ 'loss_cls_weight': 1.0,
|
|
|
+ 'loss_box_weight': 5.0,
|
|
|
+ 'loss_dfl_weight': 1.0,
|
|
|
+ # ---------------- Train config ----------------
|
|
|
+ 'trainer_type': 'rtcdet',
|
|
|
+ },
|
|
|
+
|
|
|
'rtcdet_l':{
|
|
|
# ---------------- Model config ----------------
|
|
|
## Backbone
|
yjh0410