"""
ISDNet: Integrating Shallow and Deep Networks for Efficient Ultra-high Resolution Segmentation

A standalone PyTorch implementation.
"""

from .models import ISDNet
from .datasets import FLAIRDataset
from .config import (
    DATA_ROOT,
    STDC_PRETRAIN_PATH,
    BATCH_SIZE_PER_GPU,
    NUM_WORKERS,
    BASE_LR,
    WEIGHT_DECAY,
    NUM_EPOCHS,
    NUM_CLASSES,
    CROP_SIZE,
    DOWN_RATIO,
    IGNORE_INDEX,
    SAVE_INTERVAL,
)

__version__ = "1.0.0"
__all__ = [
    "ISDNet",
    "FLAIRDataset",
]