flowdis/__init__.py

"""FlowDIS: Language-Guided Dichotomous Image Segmentation with Flow Matching"""

from flowdis.configs import configs
from flowdis.loaders import load_autoencoder, load_clip, load_t5, load_transformer
from flowdis.sampling import flowdis_predict
from flowdis.util import load_models

__all__ = [
    "configs",
    "load_autoencoder",
    "load_clip",
    "load_t5",
    "load_transformer",
    "flowdis_predict",
    "load_models",
]