File size: 810 Bytes
da3eeba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
import warnings
warnings.filterwarnings("ignore", category=UserWarning, module="mobile_sam")
from .automatic_mask_generator import SamAutomaticMaskGenerator # noqa: E402
from .build_sam import (build_sam, build_sam_vit_b, build_sam_vit_h, build_sam_vit_l, # noqa: E402
build_sam_vit_t, sam_model_registry)
from .predictor import SamPredictor # noqa: E402
__all__ = [
"build_sam",
"build_sam_vit_h",
"build_sam_vit_l",
"build_sam_vit_b",
"build_sam_vit_t",
"sam_model_registry",
"SamPredictor",
"SamAutomaticMaskGenerator",
]
|