mmsegmentation

deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py

_base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py'
model = dict(
    pretrained='torchvision://resnet101',
    backbone=dict(type='ResNet', depth=101))

data = dict(
    test=dict(
        type='CustomDataset',
        img_dir='/mnt/nfs-mnj-hot-01/tmp/msaito/ddad/train_val/GT/000000/rgb/CAMERA_09',
        img_suffix='.png',
        test_mode=True,
        classes=(
            'road', 'sidewalk', 'building', 'wall', 'fence', 'pole',
            'traffic light', 'traffic sign', 'vegetation', 'terrain', 'sky',
            'person', 'rider', 'car', 'truck', 'bus', 'train', 'motorcycle',
            'bicycle'),
        palette=[
            [128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156],
            [190, 153, 153], [153, 153, 153], [250, 170, 30], [220, 220, 0],
            [107, 142, 35], [152, 251, 152], [70, 130, 180], [220, 20, 60],
            [255, 0, 0], [0, 0, 142], [0, 0, 70], [0, 60, 100],
            [0, 80, 100], [0, 0, 230], [119, 11, 32]],
    )
)