推理提供商文件

影像分類

Hugging Face's logo
加入 Hugging Face 社群

並獲得增強的文件體驗

開始使用

影像分類

影像分類是將標籤或類別分配給整個影像的任務。預計每張影像只有一個類別。

有關影像分類任務的更多詳細資訊,請檢視其專用頁面!您將找到示例和相關材料。

推薦模型

在此處探索所有可用模型並找到最適合您的模型。

使用API

import os
from huggingface_hub import InferenceClient

client = InferenceClient(
    provider="hf-inference",
    api_key=os.environ["HF_TOKEN"],
)

output = client.image_classification("cats.jpg", model="Falconsai/nsfw_image_detection")

API規範

請求

標頭
授權 字串 身份驗證標頭,格式為'Bearer: hf_****',其中hf_****是具有“推理提供商”許可權的個人使用者訪問令牌。您可以從您的設定頁面生成一個。
有效負載
輸入* 字串 作為 base64 編碼字串的輸入影像資料。如果未提供 `parameters`,您也可以將影像資料作為原始位元組負載提供。
引數 物件
        要應用的函式 列舉 可能的值:sigmoid、softmax、none。
        top_k 整數 指定時,將輸出限制為最有可能的 K 個類別。

響應

正文
(陣列) 物件陣列 輸出是一個物件陣列。
        標籤 字串 預測的類別標籤。
        分數 數字 對應的機率。
< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.