dh-faster-rcnn_r50_fpn_1x_coco.py 845 B

1234567891011121314151617181920212223
  1. _base_ = '../faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py'
  2. model = dict(
  3. roi_head=dict(
  4. type='DoubleHeadRoIHead',
  5. reg_roi_scale_factor=1.3,
  6. bbox_head=dict(
  7. _delete_=True,
  8. type='DoubleConvFCBBoxHead',
  9. num_convs=4,
  10. num_fcs=2,
  11. in_channels=256,
  12. conv_out_channels=1024,
  13. fc_out_channels=1024,
  14. roi_feat_size=7,
  15. num_classes=80,
  16. bbox_coder=dict(
  17. type='DeltaXYWHBBoxCoder',
  18. target_means=[0., 0., 0., 0.],
  19. target_stds=[0.1, 0.1, 0.2, 0.2]),
  20. reg_class_agnostic=False,
  21. loss_cls=dict(
  22. type='CrossEntropyLoss', use_sigmoid=False, loss_weight=2.0),
  23. loss_bbox=dict(type='SmoothL1Loss', beta=1.0, loss_weight=2.0))))