votenet_head-iouloss_8xb8_scannet-3d.py 225 Bytes
Newer Older
1
_base_ = ['./votenet_8xb8_scannet-3d.py']
2
3
4
5
6
7
8

# model settings, add iou loss
model = dict(
    bbox_head=dict(
        iou_loss=dict(
            type='AxisAlignedIoULoss', reduction='sum', loss_weight=10.0 /
            3.0)))