No. Maybe? I don't know. Use this app to find out.
Use the MobileNet model to confirm your suspicions about bananas 🍌🤔
Visit the live demo here on mobile or desktop and upload an image. After a few seconds, you'll get a prediction.
Feel free to test it out with these images: Banana, Hamburger, Apple
I wanted to learn more about machine learning and how to use TensorFlow.js. I explored the TensorFlow.js website and found a pre-trained image classification model called MobileNet. I used this model to classify images of bananas and other objects.
I first had to learn how to use the URL.createObjectURL method to create a URL for the image file. After reading the TensorFlow docs, I learned that the tf.browser.fromPixels method can be used to convert an image to a tensor. The fromPixels method takes in either an ImageData, HTMLImageElement, HTMLCanvasElement, or HTMLVideoElement. I learnt the image data contained the imageElement which could be used to convert the image to a tensor.
Using the tensor, I could then pass it into the classify method of the MobileNet model. The classify method returns a promise which resolves to an array of predictions. I then used the map method to get the top predictions and determined if the predictions included the string banana.