fast_sam/__init__.py

from .fast_sam_wrapper import FastSAM
from .fast_sam_wrapper import FastSamAutomaticMaskGenerator

fast_sam_model_registry = {
    "FastSAM-x": FastSAM,
    "FastSAM-s": FastSAM,
}

__all__ = ["FastSAM", "FastSamAutomaticMaskGenerator", "fast_sam_model_registry"]