Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
images
README.md
img2num.py
img2obj.py
lenet5_data.tar
nn_img2num.py
saved_cnn_model_save.tar
test.py

README.md

Real Time Classification Video Stream Using a CIFAR100 LeNet5 Convolutional Neural Network

Aakash Pydi


In img2num.py, a LeNet5 convolution neural network is used to carry out classification for the MNIST (handwritten digits) dataset. In img2obj.py, a LeNet5 convolution neural network is used to carry out classification for the CIFAR100 dataset. Note that nn_img2num.py is also included to carry out comparison of the LeNet5 CNN model and the simple NN model from the last homework. (This was done by storing the plot details from img2num.py and then accessing that information from nn_img2num, and then subsequently plotting the required charts). The file lenet5_data.tar has the required plot information from running img2num.py.

The file saved_cnn_model_save.tar stores the evaluated CIFAR100 LeNet5 CNN model. This allows one to quickly load the model (circumventing the lengthy training time). If one wants to train the model again, simply delete this save file.

The associated results are given below.


Image to Number (MNIST Dataset)

Comparison Between LeNet5 CNN and Simple NN for MNIST Digit Classification


Image to Object (CIFAR100 Dataset)

One Object and its Caption

Continuous Captioning of Images from a Webcam

You can’t perform that action at this time.