"...pipeline_stable_diffusion_xl_instandid_img2img.py" did not exist on "a51b6cc86a5bef62283562d49497a4f3e0b134d8"
spvcnn.py 880 Bytes
Newer Older
Sun Jiahao's avatar
Sun Jiahao committed
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),
Sun Jiahao's avatar
Sun Jiahao committed
13
14
15
16
    backbone=dict(
        type='SPVCNNBackbone',
        in_channels=4,
        base_channels=32,
17
        num_stages=4,
Sun Jiahao's avatar
Sun Jiahao committed
18
19
20
21
22
23
24
25
26
27
28
29
        encoder_channels=[32, 64, 128, 256],
        decoder_channels=[256, 128, 96, 96],
        drop_ratio=0.3),
    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())