Intermediate

Machine learning

Python

Image classification

An artist’s illustration of artificial intelligence (AI). This image represents how accountability needs to be core to AI systems. It was created by Champ Panupong Techawongthawonas as part of the Visualising AI project launched by Google DeepMind.

Image Recognition

a simple image recognition application using a machine learning library and a dataset of images

An image recognition application allows users to classify images into different categories using a machine learning library and a dataset of images. In this project, you will build a simple image recognition application.

Create a Simple Image Recognition Application

Requirements

  • A way to input an image and receive a classification from the application
  • A way to train a machine learning model using a dataset of images
  • A way to use the trained model to classify new images

Bonus

  • Can you add a feature that allows users to specify their own categories and train the model on a custom dataset?
  • Can you add a feature that allows users to upload their own images for classification?

Hint

You can use a machine learning library like scikit-learn or TensorFlow to implement the image classification model for this project. You can also use a dataset of images like the ImageNet dataset to train the model.