View all @tensorflow/tfjs-automl analysis

How to use the @tensorflow/tfjs-automl.loadImageClassification function in @tensorflow/tfjs-automl

To help you get started, we’ve selected a few @tensorflow/tfjs-automl examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

tensorflow / tfjs / tfjs-automl / demo / img_classification / index.js View on Github

async function run() {
  const model = await automl.loadImageClassification(MODEL_URL);
  const image = document.getElementById('daisy');
  const predictions = await model.classify(image);

  // Show the resulting object on the page.
  const pre = document.createElement('pre');
  pre.textContent = JSON.stringify(predictions, null, 2);
  document.body.append(pre);
}