TensorFlow.js 模型

Explore pre-trained models to add computer vision, natural language processing (NLP), and other common ML tasks to your web and browser-based applications.

Vision

Analyze features in images and videos. Unlock new real-time experiences in the browser.

圖片分類

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

物件偵測

在單一影像中定位及辨識多個物件 (Coco SSD)。

語意區隔

在瀏覽器 (DeepLab) 中執行語意區隔。

Body

Detect key points and poses on the face, hands, and body with models from MediaPipe and beyond, optimized for JavaScript and Node.js.

簡易臉部偵測

使用 Single Shot Detector 架構搭配自訂編碼器 (Blazeface) 偵測圖片中的臉孔。

臉部特徵偵測

預測 486 個 3D 臉部特徵點,推論人臉表面的相似幾何。

姿勢偵測

統合姿勢偵測 API,可從三個模型中選用一個,即時偵測非典型姿勢和快速的肢體動作。

人體分節

即時區隔不同人物和身體各部位。

手勢偵測

手掌偵測工具和全手指骨追蹤模型。每偵測到一次手部,即可預測 21 個 3D 手部關鍵點。

肖像深度評估

Estimate a depth map for a single portrait image of a human.

Text

Enable NLP in your web app using the power of BERT and other Transformer encoder architectures.

自然語言問題回答

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

文字惡意指數偵測

針對留言在對話中可能造成的影響評分 (惡意指數,從「極度惡意」到「極度善意」)。

Universal Sentence Encoder

在嵌入項目中對文字做編碼處理,以便執行情緒分類和文字相似度分析等自然語言處理工作 (Universal Sentence Encoder)。

Audio

Classify audio to detect sounds and trigger an action in your web app.

語音指令辨識

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

General

Find more TensorFlow.js models that can be used out of the box.

KNN 分類演算法

可使用 KNN 分類演算法建立分類工具的公用程式。可用於遷移學習。