TensorFlow.js 模型

探索預訓練模型,將計算機視覺、自然語言處理 (NLP) 和其他常見的機器學習任務新增到您的 Web 和瀏覽器應用程式中。

視覺

分析影像和影片中的特徵。在瀏覽器中解鎖全新的即時體驗。

影像分類

使用來自 ImageNet 資料庫的標籤對影像進行分類(MobileNet)。

目標檢測

在單個影像中定位並識別多個物件(Coco SSD)。

語義分割

在瀏覽器中執行語義分割(DeepLab)。

身體

透過 MediaPipe 及其他來源的模型,檢測臉部、手部和身體的關鍵點與姿勢,並針對 JavaScript 和 Node.js 進行了最佳化。

簡易人臉檢測

使用具有自定義編碼器的單次檢測器 (SSD) 架構來檢測影像中的人臉(Blazeface)。

人臉特徵點檢測

預測 486 個 3D 人臉特徵點,以推斷人臉的近似表面幾何結構。

姿勢檢測

統一的姿勢檢測 API,可使用三種模型中的任意一種,幫助以即時效能檢測非典型姿勢和快速身體動作。

身體分割

即時分割人物及身體部位。

手勢檢測

手掌檢測器和手部骨架手指追蹤模型。為檢測到的每隻手預測 21 個 3D 手部關鍵點。

人像深度估計

為單張人物肖像影像估計深度圖。

文字

利用 BERT 和其他 Transformer 編碼器架構的強大功能,在您的 Web 應用中啟用 NLP。

自然語言問答

使用 BERT 根據給定文字段落的內容回答問題。

文字毒性檢測

對評論對對話可能產生的影響進行評分,從“非常毒性”到“非常健康”(Toxicity)。

通用句子編碼器

將文字編碼為嵌入向量,用於情感分類和文字相似度等 NLP 任務(通用句子編碼器)。

音訊

對音訊進行分類,以檢測聲音並在您的 Web 應用中觸發操作。

語音指令識別

對來自語音指令資料集的 1 秒音訊片段進行分類(speech-commands)。

通用

查詢更多可直接使用的 TensorFlow.js 模型。

KNN 分類器

使用 K-最近鄰演算法建立分類器的工具。可用於遷移學習。