ppyolo_r50vd_dcn.yml 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  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: PPYOLOFPN
  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. PPYOLOFPN:
  20. coord_conv: true
  21. drop_block: true
  22. block_size: 3
  23. keep_prob: 0.9
  24. spp: true
  25. YOLOv3Head:
  26. anchors: [[10, 13], [16, 30], [33, 23],
  27. [30, 61], [62, 45], [59, 119],
  28. [116, 90], [156, 198], [373, 326]]
  29. anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]]
  30. loss: YOLOv3Loss
  31. iou_aware: true
  32. iou_aware_factor: 0.4
  33. YOLOv3Loss:
  34. ignore_thresh: 0.7
  35. downsample: [32, 16, 8]
  36. label_smooth: false
  37. scale_x_y: 1.05
  38. iou_loss: IouLoss
  39. iou_aware_loss: IouAwareLoss
  40. IouLoss:
  41. loss_weight: 2.5
  42. loss_square: true
  43. IouAwareLoss:
  44. loss_weight: 1.0
  45. BBoxPostProcess:
  46. decode:
  47. name: YOLOBox
  48. conf_thresh: 0.01
  49. downsample_ratio: 32
  50. clip_bbox: true
  51. scale_x_y: 1.05
  52. nms:
  53. name: MatrixNMS
  54. keep_top_k: 100
  55. score_threshold: 0.01
  56. post_threshold: 0.01
  57. nms_top_k: -1
  58. background_label: -1