推理提供商文档
图像分割
加入 Hugging Face 社区
并获得增强的文档体验
开始使用
图像分割
图像分割将图像分成多个区域,其中图像中的每个像素都映射到一个对象。
有关 image-segmentation
任务的更多详细信息,请查看其专用页面!您将找到示例和相关资料。
推荐模型
- openmmlab/upernet-convnext-small:在ADE20k上训练的语义分割模型。
- facebook/mask2former-swin-large-coco-panoptic:在COCO(常见对象)数据集上训练的全景分割模型。
在这里探索所有可用模型,找到最适合您的模型。
使用API
语言
客户端
提供商
import os
from huggingface_hub import InferenceClient
client = InferenceClient(
provider="hf-inference",
api_key=os.environ["HF_TOKEN"],
)
output = client.image_segmentation("cats.jpg", model="jonathandinu/face-parsing")
API规范
请求
标头 | ||
---|---|---|
授权 | 字符串 | 身份验证头,形式为 'Bearer: hf_****' ,其中 hf_**** 是具有“推理提供商”权限的个人用户访问令牌。您可以从您的设置页面生成。 |
有效负载 | ||
---|---|---|
输入* | 字符串 | 作为 base64 编码字符串的输入图像数据。如果未提供 `parameters`,您也可以将图像数据作为原始字节负载提供。 |
参数 | 对象 | |
mask_threshold | 数字 | 将预测的掩码转换为二进制值时使用的阈值。 |
overlap_mask_area_threshold | 数字 | 用于消除小型、不连贯片段的掩码重叠阈值。 |
subtask | 枚举 | 可能的值:instance、panoptic、semantic。 |
threshold | 数字 | 用于过滤预测掩码的概率阈值。 |
响应
正文 | ||
---|---|---|
(数组) | 对象数组 | 预测的掩码/片段 |
标签 | 字符串 | 预测片段的标签。 |
mask | 字符串 | 对应的黑白图像掩码(Base64编码)。 |
score | 数字 | 模型的得分或置信度。 |