You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

gfl_r50_fpn_mstrain_2x_coco.py 788 B

2 years ago
12345678910111213141516171819202122
  1. _base_ = './gfl_r50_fpn_1x_coco.py'
  2. # learning policy
  3. lr_config = dict(step=[16, 22])
  4. runner = dict(type='EpochBasedRunner', max_epochs=24)
  5. # multi-scale training
  6. img_norm_cfg = dict(
  7. mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
  8. train_pipeline = [
  9. dict(type='LoadImageFromFile'),
  10. dict(type='LoadAnnotations', with_bbox=True),
  11. dict(
  12. type='Resize',
  13. img_scale=[(1333, 480), (1333, 800)],
  14. multiscale_mode='range',
  15. keep_ratio=True),
  16. dict(type='RandomFlip', flip_ratio=0.5),
  17. dict(type='Normalize', **img_norm_cfg),
  18. dict(type='Pad', size_divisor=32),
  19. dict(type='DefaultFormatBundle'),
  20. dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels']),
  21. ]
  22. data = dict(train=dict(pipeline=train_pipeline))

No Description

Contributors (3)