by DavidRockDesign (

Ever wanted to use image recognition, but didn’t want to be limited to just the fixed classes that come out of the box with pretrained models or cloud APIs? Today we’ll walk through how to train a ML model to recognize some fruits it’s never seen before with 95% accuracy!

A Brief Intro on Image Classification and Retraining

Original Image

Tensorflow.js is a new deep learning library that runs right in your browser. Being a machine learning and Javascript enthusiast, I immediately started working on an object detection library using Tensorflow.js after it was released.

Here, I’ll walk through how I got started, and break down the different points of…

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