| # 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. | |
| # modified by ziqi-jin | |
| # from .build_sam import ( | |
| # build_sam, | |
| # build_sam_vit_h, | |
| # build_sam_vit_l, | |
| # build_sam_vit_b, | |
| # sam_model_registry, | |
| # ) | |
| from .modeling.sam import Sam | |
| from .predictor import SamPredictor | |
| from .automatic_mask_generator import SamAutomaticMaskGenerator | |