rofunc.utils.visualab.segment.vlpart_sam_seg#

1.  Module Contents#

1.1.  Functions#

vlpart_sam_predict

show_predictions_with_masks

1.2.  API#

rofunc.utils.visualab.segment.vlpart_sam_seg.vlpart_sam_predict(image, text_prompt, vlpart_checkpoint='swinbase_part_0a0000.pth', sam_checkpoint='sam_vit_h_4b8939.pth', box_threshold=0.3, text_threshold=0.25, device='cuda')[source]#
rofunc.utils.visualab.segment.vlpart_sam_seg.show_predictions_with_masks(scores, boxes, classes, masks, text_prompt)[source]#