"lm_eval/tasks/darijammlu/darijammlu_logical_fallacies.yaml" did not exist on "5a9d5ba0234493e1d67868d8b21ae37bf0e36bc2"
paconv_ssg_8xb8-cosine-150e_s3dis-seg.py 1.71 KB
Newer Older
1
_base_ = [
2
3
    '../_base_/datasets/s3dis-seg.py', '../_base_/models/paconv_ssg.py',
    '../_base_/schedules/seg-cosine-150e.py', '../_base_/default_runtime.py'
4
5
]

ChaimZhu's avatar
ChaimZhu committed
6
7
8
9
10
11
12
13
14
15
16
# model settings
model = dict(
    decode_head=dict(
        num_classes=13, ignore_index=13,
        loss_decode=dict(class_weight=None)),  # S3DIS doesn't use class_weight
    test_cfg=dict(
        num_points=4096,
        block_size=1.0,
        sample_rate=0.5,
        use_normalized_coord=True,
        batch_size=12))
17

18
19
20
21
22
23
24
25
26
# data settings
num_points = 4096
train_pipeline = [
    dict(
        type='LoadPointsFromFile',
        coord_type='DEPTH',
        shift_height=False,
        use_color=True,
        load_dim=6,
27
        use_dim=[0, 1, 2, 3, 4, 5]),
28
29
30
31
32
    dict(
        type='LoadAnnotations3D',
        with_bbox_3d=False,
        with_label_3d=False,
        with_mask_3d=False,
33
        with_seg_3d=True),
34
    dict(type='PointSegClassMapping'),
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
    dict(
        type='IndoorPatchPointSample',
        num_points=num_points,
        block_size=1.0,
        use_normalized_coord=True,
        num_try=10000,
        enlarge_size=None,
        min_unique_num=num_points // 4,
        eps=0.0),
    dict(type='NormalizePointsColor', color_mean=None),
    dict(
        type='GlobalRotScaleTrans',
        rot_range=[0.0, 6.283185307179586],  # [0, 2 * pi]
        scale_ratio_range=[0.8, 1.2],
        translation_std=[0, 0, 0]),
    dict(
        type='RandomJitterPoints',
        jitter_std=[0.01, 0.01, 0.01],
        clip_range=[-0.05, 0.05]),
    dict(type='RandomDropPointsColor', drop_ratio=0.2),
55
    dict(type='Pack3DDetInputs', keys=['points', 'pts_semantic_mask'])
56
57
]

58
59
train_dataloader = dict(batch_size=8, dataset=dict(pipeline=train_pipeline))
val_cfg = dict(interval=1)