File size: 297 Bytes
9a84ec8
 
 
 
 
1
2
3
4
5
from .base_segmenter import BaseSegmenter
from caption_anything.utils.utils import seg_model_map

def build_segmenter(model_name, device, args=None, model=None):
        return BaseSegmenter(device, args.segmenter_checkpoint, model_name, reuse_feature=not args.disable_reuse_features, model=model)