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.
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/.