Spaces:
Runtime error
Runtime error
File size: 944 Bytes
9cc3eb2 |
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 27 28 29 30 31 32 33 34 35 36 37 38 |
from app.models.last_layer import LastLayerNeck
from app.models.model import SAMSegmentor
from app.models.sam_backbone import SAMBackbone
from app.models.sam_mask_decoder import SAMMaskDecoder
from app.models.sam_pe import SAMPromptEncoder
model = dict(
type=SAMSegmentor,
data_preprocessor=None,
backbone=dict(
type=SAMBackbone,
model_name='vit_h',
fix=True,
init_cfg=dict(
type='sam_pretrain',
checkpoint='vit_h'
)
),
neck=dict(type=LastLayerNeck),
prompt_encoder=dict(
type=SAMPromptEncoder,
model_name='vit_h',
fix=True,
init_cfg=dict(
type='sam_pretrain',
checkpoint='vit_h'
)
),
mask_decoder=dict(
type=SAMMaskDecoder,
model_name='vit_h',
fix=True,
init_cfg=dict(
type='sam_pretrain',
checkpoint='vit_h'
)
)
) |