This is an image recognition program for recognizing handwritten digits images from the MNIST dataset. Four different achitectures of the Convolution Neural Network was implemented, a modified CNN model, LeNet, VGG and a simple CNN model. The objective of this program is to evaluate these different models in terms of the learning curve, accuracy and training time in respect of the application.
- Ensure 'scikit-learn' is installed (might require a restart after installation), instructions can be found on: https://anaconda.org/anaconda/scikit-learn
- Our program does not use GPU for simplicity, thus the training time is very long.
- Open Command Prompt
- Redirect to the project folder "Handwritten-Image-Recognition-using-Convolutional-Neural-Network-models"
- Run command: python main.py
- Enter the name of the model and the number of epochs to run
- As the program is executing, the selected model will start the training and testing process. Information of the results of the training and the testing processes will appear as it continues to run.
- Wait for the simulation to finished, plots will appear showing the loss factors in the training process and the improvement of accuracy in after the testing process