Machine Learning Test 1

by codingdude

HTML

<script src="https://unpkg.com/[email protected]/dist/ml5.min.js"></script>
<img src="https://ml5js.org/docs/assets/img/bird.jpg"
     crossorigin="anonymous" id="image" width="400">
     
<p><input type="text" class="imgSrc" size="100"></p>
<p>Result: <span class="result"></span></p>
<p>Probability: <span class="probability"></span></p>

JavaScript

const classifier = ml5.imageClassifier('MobileNet', function() {
  console.log('Model Loaded!');
  predict();
});

document.getElementById("image").onload = function(){
	predict();
}

document.querySelector(".imgSrc").onchange = function(){
	document.getElementById("image").src = this.value;
}

function predict(){
  classifier.predict(image, function(err, results) {
    document.querySelector(".result").innerText = results[0].className;
    document.querySelector(".probability").innerText = results[0].probability.toFixed(4);
  });
}