fcos_r50-caffe_fpn_gn-head-center-normbbox-centeronreg-giou_1x_coco.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. _base_ = 'fcos_r50-caffe_fpn_gn-head_1x_coco.py'
  2. # model setting
  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. init_cfg=dict(
  12. type='Pretrained',
  13. checkpoint='open-mmlab://detectron2/resnet50_caffe')),
  14. bbox_head=dict(
  15. norm_on_bbox=True,
  16. centerness_on_reg=True,
  17. dcn_on_last_conv=False,
  18. center_sampling=True,
  19. conv_bias=True,
  20. loss_bbox=dict(type='GIoULoss', loss_weight=1.0)),
  21. # training and testing settings
  22. test_cfg=dict(nms=dict(type='nms', iou_threshold=0.6)))
  23. # learning rate
  24. param_scheduler = [
  25. dict(
  26. type='LinearLR',
  27. start_factor=1.0 / 3.0,
  28. by_epoch=False,
  29. begin=0,
  30. end=500),
  31. dict(
  32. type='MultiStepLR',
  33. begin=0,
  34. end=12,
  35. by_epoch=True,
  36. milestones=[8, 11],
  37. gamma=0.1)
  38. ]
  39. # optimizer
  40. optim_wrapper = dict(clip_grad=None)