by tensorflow

tensorflow / tfjs-models

Pretrained models for TensorFlow.js

7.5K Stars 2.3K Forks Last release: Not found Apache License 2.0 418 Commits 38 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

Pre-trained TensorFlow.js models

This repository hosts a set of pre-trained models that have been ported to TensorFlow.js.

The models are hosted on NPM and unpkg so they can be used in any project out of the box. They can be used directly or used in a transfer learning setting with TensorFlow.js.

To find out about APIs for models, look at the README in each of the respective directories. In general, we try to hide tensors so the API can be used by non-machine learning experts.

For those interested in contributing a model, please file a GitHub issue on tfjs to gauge interest. We are trying to add models that complement the existing set of models and can be used as building blocks in other apps.


Type Model Demo Details Install
Classify images with labels from the ImageNet database. npm i @tensorflow-models/mobilenet
live A machine learning model which allows for real-time human pose estimation in the browser. See a detailed description here. npm i @tensorflow-models/posenet
Coco SSD
Object detection model that aims to localize and identify multiple objects in a single image. Based on the TensorFlow object detection API. npm i @tensorflow-models/coco-ssd
live Real-time person and body part segmentation in the browser using TensorFlow.js. npm i @tensorflow-models/body-pix
DeepLab v3
Semantic segmentation npm i @tensorflow-models/deeplab
Speech Commands
live Classify 1 second audio snippets from the speech commands dataset. npm i @tensorflow-models/speech-commands
Universal Sentence Encoder
Encode text into a 512-dimensional embedding to be used as inputs to natural language processing tasks such as sentiment classification and textual similarity. npm i @tensorflow-models/universal-sentence-encoder
Text Toxicity
live Score the perceived impact a comment might have on a conversation, from "Very toxic" to "Very healthy". npm i @tensorflow-models/toxicity
General Utilities
KNN Classifier
This package provides a utility for creating a classifier using the K-Nearest Neighbors algorithm. Can be used for transfer learning. npm i @tensorflow-models/knn-classifier


You can run the unit tests for any of the models by running the following inside a directory:

yarn test

New models should have a test NPM script (see this

helper for reference).

To run all of the tests, you can run the following command from the root of this repo:

yarn presubmit

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.