_ / fast_sam /__init__.py
Zafaflahfksdf's picture
Upload folder using huggingface_hub
da3eeba verified
raw
history blame contribute delete
273 Bytes
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"]