TensorFlow.JS - Transfer Learning Examples

These examples show different approaches to retrain a pre-trained CNN model (MobileNet v1) with new images.

The Flowers images used in these examples, are from the TensorFlow website and are not published with this repository.

Setup

For each subfolder (train or web) in each example

Prepare the node environments:

$ npm install
# Or
$ yarn

(Re)train the models (train subfolders) by running:

$ npm run start
# Or
$ yarn run start

Run the web server script (web subfolders) for testing the trained models in your local browser:

$ npm run start
# Or
$ yarn run start

If you wan't, you can test the deployed application under https://tfjs-transferlearning.azureedge.net/.