second_hv_secfpn_8xb6-80e_kitti-3d-car.py 972 Bytes
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
_base_ = [
2
3
    '../_base_/models/second_hv_secfpn_kitti.py',
    '../_base_/datasets/kitti-3d-car.py', '../_base_/schedules/cyclic-40e.py',
zhangwenwei's avatar
zhangwenwei committed
4
5
6
    '../_base_/default_runtime.py'
]
point_cloud_range = [0, -40, -3, 70.4, 40, 1]
zhangwenwei's avatar
zhangwenwei committed
7
8
model = dict(
    bbox_head=dict(
zhangwenwei's avatar
zhangwenwei committed
9
10
        type='Anchor3DHead',
        num_classes=1,
11
        anchor_generator=dict(
zhangwenwei's avatar
zhangwenwei committed
12
            _delete_=True,
13
14
            type='Anchor3DRangeGenerator',
            ranges=[[0, -40.0, -1.78, 70.4, 40.0, -1.78]],
15
            sizes=[[3.9, 1.6, 1.56]],
16
            rotations=[0, 1.57],
17
18
19
20
21
            reshape_out=True)),
    # model training and testing settings
    train_cfg=dict(
        _delete_=True,
        assigner=dict(
22
            type='Max3DIoUAssigner',
23
24
25
26
27
28
29
30
            iou_calculator=dict(type='BboxOverlapsNearest3D'),
            pos_iou_thr=0.6,
            neg_iou_thr=0.45,
            min_pos_iou=0.45,
            ignore_iof_thr=-1),
        allowed_border=0,
        pos_weight=-1,
        debug=False))