推理提供商文件
影像分類
加入 Hugging Face 社群
並獲得增強的文件體驗
開始使用
影像分類
影像分類是將標籤或類別分配給整個影像的任務。預計每張影像只有一個類別。
有關影像分類
任務的更多詳細資訊,請檢視其專用頁面!您將找到示例和相關材料。
推薦模型
- google/vit-base-patch16-224:一個強大的影像分類模型。
- facebook/deit-base-distilled-patch16-224:一個健壯的影像分類模型。
- facebook/convnext-large-224:一個強大的影像分類模型。
在此處探索所有可用模型並找到最適合您的模型。
使用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 個類別。 |
響應
正文 | ||
---|---|---|
(陣列) | 物件陣列 | 輸出是一個物件陣列。 |
標籤 | 字串 | 預測的類別標籤。 |
分數 | 數字 | 對應的機率。 |