weijiawu's picture
Duplicate from TencentARC/Caption-Anything
0ab9a32
raw
history blame
No virus
303 Bytes
from segmenter.base_segmenter import BaseSegmenter
def build_segmenter(type, device, args=None, model=None):
if type == 'base':
return BaseSegmenter(device, args.segmenter_checkpoint, reuse_feature=not args.disable_reuse_features, model=model)
else:
raise NotImplementedError()