minkunet.py 858 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
model = dict(
    type='MinkUNet',
    data_preprocessor=dict(
        type='Det3DDataPreprocessor',
        voxel=True,
        voxel_type='minkunet',
        voxel_layer=dict(
            max_num_points=-1,
            point_cloud_range=[-100, -100, -20, 100, 100, 20],
            voxel_size=[0.05, 0.05, 0.05],
            max_voxels=(-1, -1)),
12
        max_voxels=80000),
13
14
15
16
17
    backbone=dict(
        type='MinkUNetBackbone',
        in_channels=4,
        base_channels=32,
        num_stages=4,
18
19
        encoder_channels=[32, 64, 128, 256],
        decoder_channels=[256, 128, 96, 96]),
20
21
22
23
24
25
26
27
28
    decode_head=dict(
        type='MinkUNetHead',
        channels=96,
        num_classes=19,
        dropout_ratio=0,
        loss_decode=dict(type='mmdet.CrossEntropyLoss', avg_non_ignore=True),
        ignore_index=19),
    train_cfg=dict(),
    test_cfg=dict())