RT-ODLab/config/model_config/rtdetr_config.py

# yolo-free config


rtdetr_cfg = {
    # P5
    'rtdetr_n': {
        # ---------------- Model config ----------------
        ## ------- Image Encoder -------
        ### CNN-Backbone
        'backbone': 'elannet',
        'pretrained': True,
        'bk_act': 'silu',
        'bk_norm': 'BN',
        'bk_dpw': False,
        'width': 0.25,
        'depth': 0.34,
        'stride': [8, 16, 32],  # P3, P4, P5
        'max_stride': 32,
        ### CNN-Neck
        'neck': 'sppf',
        'neck_expand_ratio': 0.5,
        'pooling_size': 5,
        'neck_act': 'silu',
        'neck_norm': 'BN',
        'neck_depthwise': False,
        ### CNN-CSFM
        'fpn': 'yolo_pafpn',
        'fpn_reduce_layer': 'conv',
        'fpn_downsample_layer': 'conv',
        'fpn_core_block': 'elanblock',
        'fpn_act': 'silu',
        'fpn_norm': 'BN',
        'fpn_depthwise': False,
        ## ------- Transformer Decoder -------
        'd_model': 256,
        'attn_type': 'mhsa',
        'num_decoder_layers': 6,
        'num_queries': 300,
        'de_dim_feedforward': 1024,
        'de_num_heads': 8,
        'de_dropout': 0.1,
        'de_act': 'silu',
        'de_norm': 'LN',
        # ---------------- Train config ----------------
        ## input
        'multi_scale': [0.5, 1.0],   # 320 -> 640
        'trans_type': 'yolov5_nano',
        # ---------------- Assignment config ----------------
        ## matcher
        'set_cost_class': 2.0,
        'set_cost_bbox': 5.0,
        'set_cost_giou': 2.0,
        # ---------------- Loss config ----------------
        ## loss weight
        'focal_alpha': 0.25,
        'loss_cls_weight': 1.0,
        'loss_box_weight': 5.0,
        'loss_giou_weight': 2.0,
        # ---------------- Train config ----------------
        ## close strong augmentation
        'no_aug_epoch': 10,
        'trainer_type': 'detr',
        ## optimizer
        'optimizer': 'adamw',
        'momentum': None,
        'weight_decay': 1e-4,
        'clip_grad': 0.1,
        ## model EMA
        'ema_decay': 0.9999,       # SGD: 0.9999;   AdamW: 0.9998
        'ema_tau': 2000,
        ## lr schedule
        'scheduler': 'linear',
        'lr0': 0.0001,             # SGD: 0.01;     AdamW: 0.001
        'lrf': 0.05,               # SGD: 0.01;     AdamW: 0.01
        'warmup_momentum': 0.8,
        'warmup_bias_lr': 0.1,
        },

}