Spaces:
Running
on
Zero
Running
on
Zero
File size: 626 Bytes
a93afca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
# 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.
from .efficient_sam import build_efficient_sam
def build_efficient_sam_vitt(checkpoint=None):
return build_efficient_sam(
encoder_patch_embed_dim=192,
encoder_num_heads=3,
checkpoint=checkpoint,
).eval()
def build_efficient_sam_vits(checkpoint=None):
return build_efficient_sam(
encoder_patch_embed_dim=384,
encoder_num_heads=6,
checkpoint=checkpoint,
).eval()
|