yolov3_darknet53_original_320e_coco_1p.yml 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '_base_/yolov3_darknet53.yml',
  5. '_base_/yolov3_reader.yml',
  6. ]
  7. snapshot_epoch: 5
  8. weights: output/yolov3_darknet53_270e_coco/model_final
  9. norm_type: bn
  10. YOLOv3Loss:
  11. ignore_thresh: 0.5
  12. downsample: [32, 16, 8]
  13. label_smooth: false
  14. worker_num: 8
  15. TrainReader:
  16. inputs_def:
  17. num_max_boxes: 50
  18. sample_transforms:
  19. - Decode: {}
  20. - RandomDistort: {}
  21. - RandomExpand: {fill_value: [123.675, 116.28, 103.53], ratio: 2.0}
  22. - RandomCrop: {}
  23. - RandomFlip: {}
  24. batch_transforms:
  25. - BatchRandomResize: {target_size: [416], random_size: True, random_interp: True, keep_ratio: False}
  26. - NormalizeBox: {}
  27. - PadBox: {num_max_boxes: 50}
  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: {anchor_masks: [[6, 7, 8], [3, 4, 5], [0, 1, 2]], anchors: [[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326]], downsample_ratios: [32, 16, 8], iou_thresh: 0.5}
  32. batch_size: 32
  33. shuffle: true
  34. drop_last: true
  35. mixup_epoch: -1
  36. use_shared_memory: true
  37. epoch: 320
  38. LearningRate:
  39. base_lr: 0.001
  40. schedulers:
  41. - !CosineDecay
  42. max_epochs: 320
  43. - !LinearWarmup
  44. start_factor: 0.
  45. epochs: 4
  46. OptimizerBuilder:
  47. optimizer:
  48. momentum: 0.9
  49. type: Momentum
  50. regularizer:
  51. factor: 0.016
  52. type: L2