TensorFlow.js models (original) (raw)

Stay organized with collections Save and categorize content based on your preferences.

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

Object detection

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

Simple face detection

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

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.

Hand pose detection

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

Text toxicity detection

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

Universal sentence encoder

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

KNN Classifier

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