ISDNet-pytorch / __init__.py
Antoine1091's picture
Upload folder using huggingface_hub
a585f5a verified
"""
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",
]