Image Classification with ML5.js

Integrating ML5.js with React App for image classification

MobileNet and DoodleNet models

const classifier = ml5.imageClassifier(classificationModel, () => {
  console.log(`Model loaded: ${classificationModel}`);
});

React.useEffect(() => {
  setDom(document.getElementById("image"));
  if (dom) {
    classifier.predict(dom, 5, (err, results) => {
      if (err) return;
      setPredictions(results);
    });
  }
}, [img, dom, classifier]);

Predictions