hv_second_secfpn_6x8_80e_kitti-3d-car.py 913 Bytes
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
2
3
4
5
6
_base_ = [
    '../_base_/models/hv_second_secfpn.py',
    '../_base_/datasets/kitti-3d-car.py', '../_base_/schedules/cyclic_40e.py',
    '../_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
15
16
            type='Anchor3DRangeGenerator',
            ranges=[[0, -40.0, -1.78, 70.4, 40.0, -1.78]],
            sizes=[[1.6, 3.9, 1.56]],
            rotations=[0, 1.57],
zhangwenwei's avatar
zhangwenwei committed
17
            reshape_out=True)))
zhangwenwei's avatar
zhangwenwei committed
18
19
# model training and testing settings
train_cfg = dict(
zhangwenwei's avatar
zhangwenwei committed
20
    _delete_=True,
zhangwenwei's avatar
zhangwenwei committed
21
22
    assigner=dict(
        type='MaxIoUAssigner',
zhangwenwei's avatar
zhangwenwei committed
23
        iou_calculator=dict(type='BboxOverlapsNearest3D'),
zhangwenwei's avatar
zhangwenwei committed
24
25
26
27
28
29
30
        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)