_base_ = 'mmdet::_base_/default_runtime.py'

custom_imports = dict(
    imports=['xdecoder'], allow_failed_imports=False)

model = dict(
    type='XDecoder',
    data_preprocessor=dict(
        type='DetDataPreprocessor',
        mean=[123.675, 116.28, 103.53],
        std=[58.395, 57.12, 57.375],
        bgr_to_rgb=True,
        pad_size_divisor=32),
    backbone=dict(type='FocalNet'),
    head=dict(
        type='XDecoderUnifiedhead',
        in_channels=(96, 192, 384, 768),
        pixel_decoder=dict(type='XTransformerEncoderPixelDecoder'),
        transformer_decoder=dict(type='XDecoderTransformerDecoder'),
        task='semseg',
    ),
    # use_thr_for_mc=True means use threshold for multi-class
    # This parameter is only used in semantic segmentation task and
    # referring semantic segmentation task.
    test_cfg=dict(mask_thr=0.5, use_thr_for_mc=True, ignore_index=255),
)

val_cfg = dict(type='ValLoop')
test_cfg = dict(type='TestLoop')
