ppyolo_r18vd.yml 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. architecture: YOLOv3
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet18_vd_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: 18
  13. variant: d
  14. return_idx: [2, 3]
  15. freeze_at: -1
  16. freeze_norm: false
  17. norm_decay: 0.
  18. PPYOLOFPN:
  19. drop_block: true
  20. block_size: 3
  21. keep_prob: 0.9
  22. conv_block_num: 0
  23. YOLOv3Head:
  24. anchor_masks: [[3, 4, 5], [0, 1, 2]]
  25. anchors: [[10, 14], [23, 27], [37, 58],
  26. [81, 82], [135, 169], [344, 319]]
  27. loss: YOLOv3Loss
  28. YOLOv3Loss:
  29. ignore_thresh: 0.7
  30. downsample: [32, 16]
  31. label_smooth: false
  32. scale_x_y: 1.05
  33. iou_loss: IouLoss
  34. IouLoss:
  35. loss_weight: 2.5
  36. loss_square: true
  37. BBoxPostProcess:
  38. decode:
  39. name: YOLOBox
  40. conf_thresh: 0.01
  41. downsample_ratio: 32
  42. clip_bbox: true
  43. scale_x_y: 1.05
  44. nms:
  45. name: MatrixNMS
  46. keep_top_k: 100
  47. score_threshold: 0.01
  48. post_threshold: 0.01
  49. nms_top_k: -1
  50. background_label: -1