TensorFlow.js 模型

探索預先訓練的模型,將電腦視覺、自然語言處理 (NLP) 和其他常見的 ML 任務新增至您的網路和瀏覽器應用程式。

視覺

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

圖片分類

使用 ImageNet 資料庫 (MobileNet) 的標籤對圖片進行分類。

物件偵測

在單一圖片中定位並識別多個物件 (Coco SSD)。

語意分割

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

身體

使用 MediaPipe 及其他模型的模型偵測臉部、手部和身體上的關鍵點和姿勢,這些模型已針對 JavaScript 和 Node.js 最佳化。

簡易臉部偵測

使用具有自訂編碼器的 Single Shot Detector 架構 (Blazeface) 偵測圖片中的臉部。

臉部地標偵測

預測 486 個 3D 臉部地標,以推斷人臉的近似表面幾何形狀。

姿勢偵測

統一姿勢偵測 API,用於使用三種模型之一,這些模型有助於偵測非典型姿勢和快速身體動作,並具有即時效能。

身體分割

即時分割人員和身體部位。

手部姿勢偵測

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

人像深度估計

估計人類單一人像圖片的深度圖。

文字

在您的 Web 應用程式中使用 BERT 和其他 Transformer 編碼器架構的功能啟用 NLP。

自然語言問題解答

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

文字毒性偵測

評估評論可能對對話產生的感知影響,從「劇毒」到「非常健康」(毒性)。

通用句子編碼器

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

音訊

分類音訊以偵測聲音並在您的 Web 應用程式中觸發動作。

語音指令辨識

從語音指令資料集 (speech-commands) 分類 1 秒音訊片段。

一般

尋找更多可直接使用的 TensorFlow.js 模型。

KNN 分類器

使用 K 最近鄰演算法建立分類器的公用程式。可用於轉移學習。