SAM提示分割

功能： 使用Segment Anything模型，输入boxes或points提示进行图片的分割, 本模型依赖segment-anything这一python库，若没装则需要在qianyi的pypi源中用pip install segment-anything安装

输入参数：

名称 类型 合法范围 缺省值 含义

图像

ColorImage

无

无

输入图片,需要RGB通道的彩色图片

提示点列表

List

无

无

(np.ndarray or None): Nx2的数组，数组每一行代表一个提示点像素位置(X, Y)

提示框列表

List

无

无

(np.ndarray or None): Nx4x2的数组，数组每一行代表一个提示框的四个点的坐标，会根据此计算出其左上角和右下角进行计算

输出参数：

名称 类型 合法范围 缺省值 含义

检测结果

DetectInstance

无

{}

返回包围框,分类,分数和polygon

参数设置：

名称 类型 合法范围 缺省值 含义

权重文件

File

['.pth']

无

模型文件，模型需要与模型类型相对应，官方模型下载路径为 http://10.10.10.98:9000/inference/sam/sam_vit_b_01ec64.pth、 http://10.10.10.98:9000/inference/sam/sam_vit_l_0b3195.pth、 http://10.10.10.98:9000/inference/sam/sam_vit_h_4b8939.pth

模型类型

String

['vit_b', 'vit_l', 'vit_h']

vit_b

模型类型

开启GPU

Bool

无

False

设置是否使用GPU做推理, 若开启需要电脑具有显卡环境

置信阈值

Float

[0.0, 1.0]

0.5

检测阈值

输出多个结果

Bool

无

0

对每一个提示，是否输出多个结果，若为True则每个提示输出3个掩膜，若为False则只输出一个

类别名

String

['0', '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']

0

可根据场景设置类别名称，以进行过滤等操作