Models

Explore pre-trained TensorFlow.js models that can be used in any project out of the box.

Image classification

Classify images with labels from the ImageNet database (MobileNet).

View code   

Object detection

Localize and identify multiple objects in a single image (Coco SSD).

View code   

Body segmentation

Segment person(s) and body parts in real-time (BodyPix).

View code   

Pose detection

Unified pose detection API for using one of three models that help detect atypical poses and fast body motions with real time performance.

View code   

Text toxicity detection

Score the perceived impact a comment may have on a conversation, from "Very toxic" to "Very healthy" (Toxicity).

View code   

Universal sentence encoder

Encode text into embeddings for NLP tasks such as sentiment classification and textual similarity (Universal Sentence Encoder).

View code   

Speech command recognition

Classify 1-second audio snippets from the speech commands dataset (speech-commands).

View code   

KNN Classifier

Utility to create a classifier using the K-Nearest-Neighbors algorithm. Can be used for transfer learning.

View code   

Simple face detection

Detect faces in images using a Single Shot Detector architecture with a custom encoder (Blazeface).

View code   

Semantic segmentation

Run semantic segmentation in the browser (DeepLab).

View code   

Face landmark detection

Predict 486 3D facial landmarks to infer the approximate surface geometry of human faces.

View code   

Hand pose detection

Palm detector and hand-skeleton finger tracking model. Predict 21 3D hand keypoints per detected hand.

View code   

Natural language question answering

Answer questions based on the content of a given passage of text using BERT.

View code   

See more on GitHub   

Get started with TensorFlow.js

Explore tutorials