_
File size: 273 Bytes
da3eeba
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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"]