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	live	Classify images with labels from the ImageNet database.	`npm i @tensorflow-models/mobilenet`
	MobileNet	source	Classify images with labels from the ImageNet database.	`npm i @tensorflow-models/mobilenet`
	Hand	live	Real-time hand pose detection in the browser using TensorFlow.js.	`npm i @tensorflow-models/hand-pose-detection`
	Hand	source		`npm i @tensorflow-models/hand-pose-detection`
	Pose	live	An API for real-time human pose detection in the browser.	`npm i @tensorflow-models/pose-detection`
	Pose	source	An API for real-time human pose detection in the browser.	`npm i @tensorflow-models/pose-detection`
	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`
	Coco SSD	source		`npm i @tensorflow-models/coco-ssd`
	DeepLab v3		Semantic segmentation	`npm i @tensorflow-models/deeplab`
	DeepLab v3	source	Semantic segmentation	`npm i @tensorflow-models/deeplab`
	Face Landmark Detection	live	Real-time 3D facial landmarks detection to infer the approximate surface geometry of a human face	`npm i @tensorflow-models/face-landmarks-detection`
	Face Landmark Detection	source		`npm i @tensorflow-models/face-landmarks-detection`
Audio	Speech Commands	live	Classify 1 second audio snippets from the speech commands dataset.	`npm i @tensorflow-models/speech-commands`
Audio	Speech Commands	source		`npm i @tensorflow-models/speech-commands`
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`
	Universal Sentence Encoder	source		`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`
	Text Toxicity	source		`npm i @tensorflow-models/toxicity`
Depth Estimation	Portrait Depth	live	Estimate per-pixel depth (the distance to the camera center) for a single portrait image, which can be further used for creative applications such as 3D photo and relighting.	`npm i @tensorflow-models/depth-estimation`
Depth Estimation	Portrait Depth	source		`npm i @tensorflow-models/depth-estimation`
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`
General Utilities	KNN Classifier	source		`npm i @tensorflow-models/knn-classifier`