Skip to content
Pretrained models for TensorFlow.js
TypeScript JavaScript HTML Python Shell CSS
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.tslint Add linting rules for tfjs-models. (#333) Oct 25, 2019
.vscode Generalize KNN Image Classifier to KNN Classifier, taking an arbitrar… Jun 19, 2018
body-pix Add publish-npm, make-version, tag-version scripts to models. (#369) Dec 6, 2019
coco-ssd Update lock files. (#371) Dec 6, 2019
deeplab Add publish-npm, make-version, tag-version scripts to models. (#369) Dec 6, 2019
knn-classifier Update lock files. (#371) Dec 6, 2019
mobilenet Update lock files. (#371) Dec 6, 2019
posenet Update lock files. (#371) Dec 6, 2019
scripts Remove for-publish in build-npm (#370) Dec 6, 2019
speech-commands Add publish-npm, make-version, tag-version scripts to models. (#369) Dec 6, 2019
toxicity Update lock files. (#371) Dec 6, 2019
universal-sentence-encoder Update lock files. (#371) Dec 6, 2019
.gitignore Check in a script that diffs each folder against master. (#329) Oct 22, 2019
CONTRIBUTING.md Add license and contributing Apr 30, 2018
LICENSE Add license and contributing Apr 30, 2018
README.md Add DeepLab (#229) Jul 29, 2019
cloudbuild.yml Add cloud build scripts to run individual cloudbuild.yml's (#331) Oct 22, 2019
package.json Update deeplab model urls to tfhub Oct 31, 2019
presubmit.ts Add publish-npm, make-version, tag-version scripts to models. (#369) Dec 6, 2019
run_python_tests.sh Add publish-npm, make-version, tag-version scripts to models. (#369) Dec 6, 2019
test_util.ts Add publish-npm, make-version, tag-version scripts to models. (#369) Dec 6, 2019
tslint.json Add linting rules for tfjs-models. (#333) Oct 25, 2019
yarn.lock save Oct 31, 2019

README.md

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.

Models

Type Model Demo Details Install
Images
MobileNet
Classify images with labels from the ImageNet database. npm i @tensorflow-models/mobilenet
source
PoseNet
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
source
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
source
BodyPix
live Real-time person and body part segmentation in the browser using TensorFlow.js. npm i @tensorflow-models/body-pix
source
DeepLab v3
Semantic segmentation npm i @tensorflow-models/deeplab
source
Audio
Speech Commands
live Classify 1 second audio snippets from the speech commands dataset. npm i @tensorflow-models/speech-commands
source
Text
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
source
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
source
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
source

Development

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 package.json and run_tests.ts helper for reference).

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

yarn presubmit

You can’t perform that action at this time.