1234567891011121314151617181920212223 |
- _base_ = '../faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py'
- model = dict(
- roi_head=dict(
- type='DoubleHeadRoIHead',
- reg_roi_scale_factor=1.3,
- bbox_head=dict(
- _delete_=True,
- type='DoubleConvFCBBoxHead',
- num_convs=4,
- num_fcs=2,
- in_channels=256,
- conv_out_channels=1024,
- fc_out_channels=1024,
- roi_feat_size=7,
- num_classes=80,
- bbox_coder=dict(
- type='DeltaXYWHBBoxCoder',
- target_means=[0., 0., 0., 0.],
- target_stds=[0.1, 0.1, 0.2, 0.2]),
- reg_class_agnostic=False,
- loss_cls=dict(
- type='CrossEntropyLoss', use_sigmoid=False, loss_weight=2.0),
- loss_bbox=dict(type='SmoothL1Loss', beta=1.0, loss_weight=2.0))))
|