ppyoloe_crn_s_80e_smoking_visdrone.yml 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. _BASE_: [
  2. '../runtime.yml',
  3. '../ppyoloe/_base_/optimizer_300e.yml',
  4. '../ppyoloe/_base_/ppyoloe_crn.yml',
  5. '../ppyoloe/_base_/ppyoloe_reader.yml',
  6. ]
  7. log_iter: 100
  8. snapshot_epoch: 10
  9. weights: output/ppyoloe_crn_s_80e_smoking_visdrone/model_final
  10. pretrain_weights: https://paddledet.bj.bcebos.com/models/ppyoloe_crn_s_80e_visdrone.pdparams
  11. depth_mult: 0.33
  12. width_mult: 0.50
  13. TrainReader:
  14. batch_size: 16
  15. LearningRate:
  16. base_lr: 0.01
  17. epoch: 80
  18. LearningRate:
  19. base_lr: 0.01
  20. schedulers:
  21. - !CosineDecay
  22. max_epochs: 80
  23. - !LinearWarmup
  24. start_factor: 0.
  25. epochs: 1
  26. PPYOLOEHead:
  27. static_assigner_epoch: -1
  28. metric: COCO
  29. num_classes: 1
  30. TrainDataset:
  31. !COCODataSet
  32. image_dir: ""
  33. anno_path: smoking_train_cocoformat.json
  34. dataset_dir: dataset/smoking
  35. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  36. EvalDataset:
  37. !COCODataSet
  38. image_dir: ""
  39. anno_path: smoking_test_cocoformat.json
  40. dataset_dir: dataset/smoking
  41. TestDataset:
  42. !ImageFolder
  43. anno_path: smoking_test_cocoformat.json
  44. dataset_dir: dataset/smoking