faster-rcnn_x101-32x8d_fpn_ms-3x_coco.py 784 B

1234567891011121314151617181920212223
  1. _base_ = ['../common/ms_3x_coco.py', '../_base_/models/faster-rcnn_r50_fpn.py']
  2. model = dict(
  3. # ResNeXt-101-32x8d model trained with Caffe2 at FB,
  4. # so the mean and std need to be changed.
  5. data_preprocessor=dict(
  6. type='DetDataPreprocessor',
  7. mean=[103.530, 116.280, 123.675],
  8. std=[57.375, 57.120, 58.395],
  9. bgr_to_rgb=False,
  10. pad_size_divisor=32),
  11. backbone=dict(
  12. type='ResNeXt',
  13. depth=101,
  14. groups=32,
  15. base_width=8,
  16. num_stages=4,
  17. out_indices=(0, 1, 2, 3),
  18. frozen_stages=1,
  19. norm_cfg=dict(type='BN', requires_grad=False),
  20. style='pytorch',
  21. init_cfg=dict(
  22. type='Pretrained',
  23. checkpoint='open-mmlab://detectron2/resnext101_32x8d')))