ppyoloe_crn_s_80e_visdrone.yml 853 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. _BASE_: [
  2. '../../datasets/visdrone_detection.yml',
  3. '../../runtime.yml',
  4. '../../ppyoloe/_base_/optimizer_300e.yml',
  5. '../../ppyoloe/_base_/ppyoloe_crn.yml',
  6. '../../ppyoloe/_base_/ppyoloe_reader.yml',
  7. ]
  8. log_iter: 100
  9. snapshot_epoch: 10
  10. weights: output/ppyoloe_crn_s_80e_visdrone/model_final
  11. pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_s_300e_coco.pdparams
  12. depth_mult: 0.33
  13. width_mult: 0.50
  14. TrainReader:
  15. batch_size: 8
  16. EvalReader:
  17. batch_size: 1
  18. TestReader:
  19. batch_size: 1
  20. fuse_normalize: True
  21. epoch: 80
  22. LearningRate:
  23. base_lr: 0.01
  24. schedulers:
  25. - !CosineDecay
  26. max_epochs: 96
  27. - !LinearWarmup
  28. start_factor: 0.
  29. epochs: 1
  30. PPYOLOEHead:
  31. static_assigner_epoch: -1
  32. nms:
  33. name: MultiClassNMS
  34. nms_top_k: 10000
  35. keep_top_k: 500
  36. score_threshold: 0.01
  37. nms_threshold: 0.6