Skip to content

Instantly share code, notes, and snippets.

@vabarbosa

vabarbosa/load-model-tfjs-0.x.x.js

Last active March 11, 2019 20:28
Show Gist options
  • Save vabarbosa/01deb82adbf5e48fe6655a2d43a10e64 to your computer and use it in GitHub Desktop.
Save vabarbosa/01deb82adbf5e48fe6655a2d43a10e64 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load-model-tfjs-0.x.x.js
// tensorflow.js 0.x.x
const MODEL_URL = '/model/tensorflowjs_model.pb'
const WEIGHTS_URL = '/model/weights_manifest.json'
// https://js.tensorflow.org/api/0.15.1/#loadFrozenModel
const model = await tf.loadFrozenModel(MODEL_URL, WEIGHTS_URL)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment