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.

solo_r50_fpn_3x_coco.py 942 B

2 years ago
12345678910111213141516171819202122232425262728
  1. _base_ = './solo_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, with_mask=True),
  7. dict(
  8. type='Resize',
  9. img_scale=[(1333, 800), (1333, 768), (1333, 736), (1333, 704),
  10. (1333, 672), (1333, 640)],
  11. multiscale_mode='value',
  12. keep_ratio=True),
  13. dict(type='RandomFlip', flip_ratio=0.5),
  14. dict(type='Normalize', **img_norm_cfg),
  15. dict(type='Pad', size_divisor=32),
  16. dict(type='DefaultFormatBundle'),
  17. dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels', 'gt_masks']),
  18. ]
  19. data = dict(train=dict(pipeline=train_pipeline))
  20. lr_config = dict(
  21. policy='step',
  22. warmup='linear',
  23. warmup_iters=500,
  24. warmup_ratio=1.0 / 3,
  25. step=[27, 33])
  26. runner = dict(type='EpochBasedRunner', max_epochs=36)

No Description

Contributors (3)