File size: 511 Bytes
a585f5a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | """
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",
]
|