18 lines (13 loc) · 853 Bytes

Project Classification + X

See this example live!

This is a template website to host a TensorFlow JS (TFJS) version of image classification models online.

To host your own classification + X website:

Copy this repository to your own repository
Place the TFJS model JSON and shards to the model_tfjs/ folder
Open model_tfjs/model.json and replace instances of "Functional" with "Model"
Overwrite the CLASSES_NAMES list in index.js
Make sure the image normalization steps in index.js is the same as training
Modify index.html as desired
Host on GitHub Pages

Learn more about exporting models from TensorFlow to TFJS here.

See more TFJS examples here.