ppyolov2_r50vd_dcn.yml 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. architecture: YOLOv3
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_pretrained.pdparams
  3. norm_type: sync_bn
  4. use_ema: true
  5. ema_decay: 0.9998
  6. YOLOv3:
  7. backbone: ResNet
  8. neck: PPYOLOPAN
  9. yolo_head: YOLOv3Head
  10. post_process: BBoxPostProcess
  11. ResNet:
  12. depth: 50
  13. variant: d
  14. return_idx: [1, 2, 3]
  15. dcn_v2_stages: [3]
  16. freeze_at: -1
  17. freeze_norm: false
  18. norm_decay: 0.
  19. PPYOLOPAN:
  20. drop_block: true
  21. block_size: 3
  22. keep_prob: 0.9
  23. spp: true
  24. YOLOv3Head:
  25. anchors: [[10, 13], [16, 30], [33, 23],
  26. [30, 61], [62, 45], [59, 119],
  27. [116, 90], [156, 198], [373, 326]]
  28. anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]]
  29. loss: YOLOv3Loss
  30. iou_aware: true
  31. iou_aware_factor: 0.5
  32. YOLOv3Loss:
  33. ignore_thresh: 0.7
  34. downsample: [32, 16, 8]
  35. label_smooth: false
  36. scale_x_y: 1.05
  37. iou_loss: IouLoss
  38. iou_aware_loss: IouAwareLoss
  39. IouLoss:
  40. loss_weight: 2.5
  41. loss_square: true
  42. IouAwareLoss:
  43. loss_weight: 1.0
  44. BBoxPostProcess:
  45. decode:
  46. name: YOLOBox
  47. conf_thresh: 0.01
  48. downsample_ratio: 32
  49. clip_bbox: true
  50. scale_x_y: 1.05
  51. nms:
  52. name: MatrixNMS
  53. keep_top_k: 100
  54. score_threshold: 0.01
  55. post_threshold: 0.01
  56. nms_top_k: -1
  57. background_label: -1