Download
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. New models should have a test NPM script. You can run the unit tests for any of the models by running "yarn test" inside a directory. Use off-the-shelf JavaScript models or convert Python TensorFlow models to run in the browser or under Node.js. Build and train models directly in JavaScript using flexible and intuitive APIs. Develop ML models in JavaScript, and use ML directly in the browser or in Node.js.
Features
- Pre-trained, out-of-the-box models for common use cases
- Explore pre-trained TensorFlow.js models that can be used in any project out of the box
- Retrain pre-existing ML models using your own data
- Build and train models directly in JavaScript using flexible and intuitive APIs
- Localize and identify multiple objects in a single image
- Classify images with labels from the ImageNet database
Project Samples
