22 lines (15 loc) · 1 KB

AI4ALL Project Classification + X

@AI4ALL CMU 2022

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

See a slides tutorial with screenshots here.

Learn more about exporting models from TensorFlow to TFJS here.

See more TFJS examples here.