votenet_iouloss_8x8_scannet-3d-18class.py 232 Bytes
Newer Older
1
2
3
4
5
6
7
8
_base_ = ['./votenet_8x8_scannet-3d-18class.py']

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