ppyolo_mbv3_small_coco.yml 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. './_base_/ppyolo_mbv3_small.yml',
  5. './_base_/optimizer_1x.yml',
  6. './_base_/ppyolo_reader.yml',
  7. ]
  8. snapshot_epoch: 10
  9. weights: output/ppyolo_mbv3_small_coco/model_final
  10. TrainReader:
  11. inputs_def:
  12. num_max_boxes: 90
  13. sample_transforms:
  14. - Decode: {}
  15. - Mixup: {alpha: 1.5, beta: 1.5}
  16. - RandomDistort: {}
  17. - RandomExpand: {fill_value: [123.675, 116.28, 103.53]}
  18. - RandomCrop: {}
  19. - RandomFlip: {}
  20. batch_transforms:
  21. - BatchRandomResize:
  22. target_size: [224, 256, 288, 320, 352, 384, 416, 448, 480, 512]
  23. random_size: True
  24. random_interp: True
  25. keep_ratio: False
  26. - NormalizeBox: {}
  27. - PadBox: {num_max_boxes: 90}
  28. - BboxXYXY2XYWH: {}
  29. - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
  30. - Permute: {}
  31. - Gt2YoloTarget:
  32. anchor_masks: [[3, 4, 5], [0, 1, 2]]
  33. anchors: [[11, 18], [34, 47], [51, 126], [115, 71], [120, 195], [254, 235]]
  34. downsample_ratios: [32, 16]
  35. iou_thresh: 0.25
  36. num_classes: 80
  37. batch_size: 32
  38. mixup_epoch: 200
  39. shuffle: true
  40. EvalReader:
  41. sample_transforms:
  42. - Decode: {}
  43. - Resize: {target_size: [320, 320], keep_ratio: False, interp: 2}
  44. - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
  45. - Permute: {}
  46. batch_size: 8
  47. TestReader:
  48. inputs_def:
  49. image_shape: [3, 320, 320]
  50. sample_transforms:
  51. - Decode: {}
  52. - Resize: {target_size: [320, 320], keep_ratio: False, interp: 2}
  53. - NormalizeImage: {mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225], is_scale: True}
  54. - Permute: {}
  55. batch_size: 1
  56. epoch: 270
  57. LearningRate:
  58. base_lr: 0.005
  59. schedulers:
  60. - !PiecewiseDecay
  61. gamma: 0.1
  62. milestones:
  63. - 162
  64. - 216
  65. - !LinearWarmup
  66. start_factor: 0.
  67. steps: 4000
  68. OptimizerBuilder:
  69. optimizer:
  70. momentum: 0.9
  71. type: Momentum
  72. regularizer:
  73. factor: 0.0005
  74. type: L2