| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- architecture: PPYOLOEWithAuxHead
- norm_type: sync_bn
- use_ema: true
- ema_decay: 0.9998
- ema_black_list: ['proj_conv.weight']
- custom_black_list: ['reduce_mean']
- PPYOLOEWithAuxHead:
- backbone: CSPResNet
- neck: CustomCSPPAN
- yolo_head: PPYOLOEHead
- aux_head: SimpleConvHead
- post_process: ~
- CSPResNet:
- layers: [3, 6, 6, 3]
- channels: [64, 128, 256, 512, 1024]
- return_idx: [1, 2, 3]
- use_large_stem: True
- use_alpha: True
- CustomCSPPAN:
- out_channels: [384, 384, 384]
- stage_num: 1
- block_num: 3
- act: 'swish'
- spp: true
- SimpleConvHead:
- feat_in: 288
- feat_out: 288
- num_convs: 1
- fpn_strides: [32, 16, 8]
- norm_type: 'gn'
- act: 'LeakyReLU'
- reg_max: 16
- PPYOLOEHead:
- fpn_strides: [32, 16, 8]
- grid_cell_scale: 5.0
- grid_cell_offset: 0.5
- static_assigner_epoch: 100
- use_varifocal_loss: True
- loss_weight: {class: 1.0, iou: 2.5, dfl: 0.5}
- attn_conv: 'repvgg' #
- static_assigner:
- name: ATSSAssigner
- topk: 9
- assigner:
- name: TaskAlignedAssigner
- topk: 13
- alpha: 1.0
- beta: 6.0
- is_close_gt: True #
- nms:
- name: MultiClassNMS
- nms_top_k: 1000
- keep_top_k: 300
- score_threshold: 0.01
- nms_threshold: 0.7
|
