ppyolo_r50vd_dcn_1x_visdrone.yml 996 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '../ppyolo/_base_/ppyolo_r50vd_dcn.yml',
  5. '../ppyolo/_base_/optimizer_1x.yml',
  6. '../ppyolo/_base_/ppyolo_reader.yml',
  7. ]
  8. snapshot_epoch: 8
  9. use_ema: true
  10. weights: output/ppyolo_r50vd_dcn_1x_visdrone/model_final
  11. epoch: 192
  12. LearningRate:
  13. base_lr: 0.005
  14. schedulers:
  15. - !PiecewiseDecay
  16. gamma: 0.1
  17. milestones:
  18. - 153
  19. - 173
  20. - !LinearWarmup
  21. start_factor: 0.
  22. steps: 4000
  23. OptimizerBuilder:
  24. optimizer:
  25. momentum: 0.9
  26. type: Momentum
  27. regularizer:
  28. factor: 0.0005
  29. type: L2
  30. metric: COCO
  31. num_classes: 9
  32. TrainDataset:
  33. !COCODataSet
  34. image_dir: train
  35. anno_path: annotations/train.json
  36. dataset_dir: dataset/VisDrone2019_coco
  37. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  38. EvalDataset:
  39. !COCODataSet
  40. image_dir: val
  41. anno_path: annotations/val.json
  42. dataset_dir: dataset/VisDrone2019_coco
  43. TestDataset:
  44. !ImageFolder
  45. anno_path: annotations/val.json