fcos_r50-dcn-caffe_fpn_gn-head-center-normbbox-centeronreg-giou_1x_coco.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. _base_ = 'fcos_r50-caffe_fpn_gn-head_1x_coco.py'
  2. # model settings
  3. model = dict(
  4. data_preprocessor=dict(
  5. type='DetDataPreprocessor',
  6. mean=[103.530, 116.280, 123.675],
  7. std=[1.0, 1.0, 1.0],
  8. bgr_to_rgb=False,
  9. pad_size_divisor=32),
  10. backbone=dict(
  11. dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False),
  12. stage_with_dcn=(False, True, True, True),
  13. init_cfg=dict(
  14. type='Pretrained',
  15. checkpoint='open-mmlab://detectron2/resnet50_caffe')),
  16. bbox_head=dict(
  17. norm_on_bbox=True,
  18. centerness_on_reg=True,
  19. dcn_on_last_conv=True,
  20. center_sampling=True,
  21. conv_bias=True,
  22. loss_bbox=dict(type='GIoULoss', loss_weight=1.0)),
  23. # training and testing settings
  24. test_cfg=dict(nms=dict(type='nms', iou_threshold=0.6)))
  25. # learning rate
  26. param_scheduler = [
  27. dict(
  28. type='LinearLR',
  29. start_factor=1.0 / 3.0,
  30. by_epoch=False,
  31. begin=0,
  32. end=500),
  33. dict(
  34. type='MultiStepLR',
  35. begin=0,
  36. end=12,
  37. by_epoch=True,
  38. milestones=[8, 11],
  39. gamma=0.1)
  40. ]
  41. # optimizer
  42. optim_wrapper = dict(clip_grad=None)