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.

vfnet_r50_fpn_mstrain_2x_coco.py 1.3 kB

2 years ago
123456789101112131415161718192021222324252627282930313233343536373839
  1. _base_ = './vfnet_r50_fpn_1x_coco.py'
  2. img_norm_cfg = dict(
  3. mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
  4. train_pipeline = [
  5. dict(type='LoadImageFromFile'),
  6. dict(type='LoadAnnotations', with_bbox=True),
  7. dict(
  8. type='Resize',
  9. img_scale=[(1333, 480), (1333, 960)],
  10. multiscale_mode='range',
  11. keep_ratio=True),
  12. dict(type='RandomFlip', flip_ratio=0.5),
  13. dict(type='Normalize', **img_norm_cfg),
  14. dict(type='Pad', size_divisor=32),
  15. dict(type='DefaultFormatBundle'),
  16. dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels']),
  17. ]
  18. test_pipeline = [
  19. dict(type='LoadImageFromFile'),
  20. dict(
  21. type='MultiScaleFlipAug',
  22. img_scale=(1333, 800),
  23. flip=False,
  24. transforms=[
  25. dict(type='Resize', keep_ratio=True),
  26. dict(type='RandomFlip'),
  27. dict(type='Normalize', **img_norm_cfg),
  28. dict(type='Pad', size_divisor=32),
  29. dict(type='DefaultFormatBundle'),
  30. dict(type='Collect', keys=['img']),
  31. ])
  32. ]
  33. data = dict(
  34. train=dict(pipeline=train_pipeline),
  35. val=dict(pipeline=test_pipeline),
  36. test=dict(pipeline=test_pipeline))
  37. # learning policy
  38. lr_config = dict(step=[16, 22])
  39. runner = dict(type='EpochBasedRunner', max_epochs=24)

No Description

Contributors (3)