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 estimation

Estimate human poses in real-time (PoseNet).

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  

Sentence encoding

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  

See more on GitHub  

Get started with TensorFlow.js

Explore tutorials