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.

rpn_r50_caffe_c4_1x_coco.py 1.4 kB

2 years ago
1234567891011121314151617181920212223242526272829303132333435363738
  1. _base_ = [
  2. '../_base_/models/rpn_r50_caffe_c4.py',
  3. '../_base_/datasets/coco_detection.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. # dataset settings
  7. img_norm_cfg = dict(
  8. mean=[103.530, 116.280, 123.675], std=[1.0, 1.0, 1.0], to_rgb=False)
  9. train_pipeline = [
  10. dict(type='LoadImageFromFile'),
  11. dict(type='LoadAnnotations', with_bbox=True, with_label=False),
  12. dict(type='Resize', img_scale=(1333, 800), 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']),
  18. ]
  19. test_pipeline = [
  20. dict(type='LoadImageFromFile'),
  21. dict(
  22. type='MultiScaleFlipAug',
  23. img_scale=(1333, 800),
  24. flip=False,
  25. transforms=[
  26. dict(type='Resize', keep_ratio=True),
  27. dict(type='RandomFlip'),
  28. dict(type='Normalize', **img_norm_cfg),
  29. dict(type='Pad', size_divisor=32),
  30. dict(type='ImageToTensor', keys=['img']),
  31. dict(type='Collect', keys=['img']),
  32. ])
  33. ]
  34. data = dict(
  35. train=dict(pipeline=train_pipeline),
  36. val=dict(pipeline=test_pipeline),
  37. test=dict(pipeline=test_pipeline))
  38. evaluation = dict(interval=1, metric='proposal_fast')

No Description

Contributors (3)