ml5.js: Image Classification with MobileNet

ml5.js is machine learning for the web in your web browser.

Through some clever and exciting advancements, the folks building TensorFlow.js figured out that it is possible to use the web browser’s built in graphics processing unit (GPU) to do calculations that would otherwise run very slowly using central processing unit (CPU).

Image for post
Image for post
puffin

I used the “pre-trained” MobileNet model to classify the content of an image. It predicts a King Penguin with 0.05 confidence because it has not learned yet about Puffins. It returns 3 possible results with a confidence number.

correctly identifying king penguin

https://github.com/mruanova/ml5/tree/master/image-prediction

Written by

Welcome to my Data Science blog. Please visit my career portfolio at https://mruanova.com 🚀🌎

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store