rofunc.utils.visualab.segment.efficient_sam_seg#

1.  Module Contents#

1.1.  Functions#

efficient_sam_generate

Generate all masks automatically with EfficientSAM.

efficient_sam_predict

Use mouse to select points or a box, and segment the object with prompt in the image.

1.2.  API#

rofunc.utils.visualab.segment.efficient_sam_seg.efficient_sam_generate(image, efficient_sam_checkpoint='efficientsam_s_gpu.jit')[source]#

Generate all masks automatically with EfficientSAM.

Parameters:

  • image –

  • efficient_sam_checkpoint –

Returns:

rofunc.utils.visualab.segment.efficient_sam_seg.efficient_sam_predict(image, use_point=False, use_box=False, efficient_sam_checkpoint='efficientsam_s_gpu.jit')[source]#

Use mouse to select points or a box, and segment the object with prompt in the image.

Parameters:

  • image – image to segment

  • use_point – whether to use pos/neg points to segment

  • use_box – whether to use a box to segment

  • efficient_sam_checkpoint – checkpoint path of sam model

Returns: