# yolov8 config

yolov8_cfg = {
    # input
    'trans_type': 'yolov5',
    'multi_scale': [0.5, 1.0],
    # model
    'backbone': 'elan_cspnet',
    'pretrained': True,
    'bk_act': 'silu',
    'bk_norm': 'BN',
    'bk_dpw': False,
    'width': 1.0,
    'depth': 1.0,
    'ratio': 1.0,
    'stride': [8, 16, 32],  # P3, P4, P5
    # neck
    'neck': 'sppf',
    'expand_ratio': 0.5,
    'pooling_size': 5,
    'neck_act': 'silu',
    'neck_norm': 'BN',
    'neck_depthwise': False,
    # fpn
    'fpn': 'yolov8_pafpn',
    '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,
    'reg_max': 16,
    # matcher
    'matcher': {'topk': 10,
                'alpha': 0.5,
                'beta': 6.0},
    # loss weight
    'cls_loss': 'bce', # vfl (optional)
    'loss_cls_weight': 0.5,
    'loss_iou_weight': 7.5,
    'loss_dfl_weight': 1.5,
    # training configuration
    'no_aug_epoch': 10,
    # optimizer
    'optimizer': 'sgd',      # optional: sgd, adamw
    'momentum': 0.937,         # SGD: 0.937;    AdamW: invalid
    'weight_decay': 5e-4,      # SGD: 5e-4;     AdamW: 5e-2
    'clip_grad': 10,           # SGD: 10.0;     AdamW: -1
    # model EMA
    'ema_decay': 0.9999,       # SGD: 0.9999;   AdamW: 0.9998
    'ema_tau': 2000,
    # lr schedule
    'scheduler': 'linear',
    'lr0': 0.01,              # SGD: 0.01;     AdamW: 0.004
    'lrf': 0.01,               # SGD: 0.01;     AdamW: 0.05
    'warmup_momentum': 0.8,
    'warmup_bias_lr': 0.1,
}