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 |
|---|---|---|---|---|
| Images | MobileNet |
Classify images with labels from the ImageNet database. | npm i @tensorflow-models/mobilenet |
|
| source | ||||
HandPose |
live | Real-time hand pose detection in the browser using TensorFlow.js. | npm i @tensorflow-models/handpose |
|
| 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 | ||||
BlazeFace |
live | Real-time rapid Face detection in the browser using TensorFlow.js. | npm i @tensorflow-models/blazeface< |