Skip to content
Snippets Groups Projects
Select Git revision
  • b811f434b5ebf1b54ee143789ef45bd2fa63c787
  • main default protected
2 results

image-classification

Arab Myla's avatar
Arab Myla authored
b811f434
History

Image classification

MOD 4.6 Deep Learning & Artificial Intelligence

Description

The objective of this tutorial is to write a complete image classification program in Python. Two classification models will be successively developed and tested: k-nearest neighbors (KNN) and neural networks (NN).

Requirements

In this tutorial we use Python 3.7 or higher and the library numpy.

Usage

The image database used for the experiments is CIFAR-10 which consists of 60 000 color images of size 32x32 divided into 10 classes (plane, car, bird, cat, ...). This database can be obtained at the address https://www.cs.toronto.edu/~kriz/cifar.html where are also given the indications to read the data.

The python file named read_cifar.py contains the functions needed to read the data.

The python file named knn.py contains the functions developping and testing the k-nearest neighbors classification model.

The python file named mlp.py contains the functions developping and testing the multilayer perceptron neural networks classification model.


Author

Myla Arab

License

MIT License