antoniorochaaz/cnn-project

Personal project for developing skills in python and testing different Convolutional Neural Network (CNN) architectures in different tasks.

CNN Project

Introduction

Despite having previous experiences in developing CNNs in PyTorch, I've always felt overwhelmed by the number of different of ways that one can combine convolutional, max-pooling and linear layers - with different kernel sizes and padding sizes, strides, dilation and feature numbers.

My objective is to develop skills in python and get used to using GitHub while exploring a variety of CNN architectures to find in practice which ones work best for different applications. This project will also serve as a portfolio.

The code's documentation is done with Sphinx and can be found here.

Note: Work is discontinued. More computational power would be required to improve models.

The pygame interface works and can be run through the command line with python -m main.py. The conda environment used can be found in the environment.yml.

Dataset

The first dataset I've chosen to use is the HASYv2 dataset, because it has many more classes and symbols than other symbol recognition datasets such as MNIST, and the final models could possibly be adapted in the future for translating handwritten equations (even if they are handwritten through a mouse pad of sorts) into LaTeX equations.

This also inspires me to develop some kind of application where the user can draw symbols in a 32x32 pixel grid with its mouse, and a trained net will try to guess it at the same time. The user can then add it live to the LaTeX equation. This application idea draws inspiration from Google's Quick, Draw!.

Update 14/11/2021: I've found out through a friend that a website already exists for this: (http://detexify.kirelabs.org/symbols.html). I might try using their dataset in the future

Note: In order to work with the dataset on Google Colab, I've tried uploading the data to Google Drive, which didn't work because it has trouble dealing with such a big amount of data (even unzipping the files directly from Colab didn't work). Because of this, I've organized them in a datasets.HASYv2Dataset class. However, it also can't be uploaded here because it exceeds 100MB. If you wish to use the codes presented here, you need to unzip the dataset (which can be found here) in the _Data folder (creating a HASYv2 folder with all of its contents).

Current State of Development

Note: With the return of my classes, development will be slowed down significantly in the next few months.

Base classes and a few models have already been defined, but development is hindered by the time it takes to train models. Currently, using Google Colab's GPU acceleration, training on one fold takes several hours to complete (which basically uses up the daily available GPU runtime). Training a model with all 10 folds would thus take up to 10 days for each model. Development should thus shift to implementations with PyTorch Lightning and PyTorch/XLA (see Objectives), that could allow for multicore TPU training in Colab, speeding up the process.

In terms of the Pygame implementation (see "pygame-tests" branch), much has yet to be done and improved, but the base window is about what I had in mind. I have understood a little bit better how Pygame works, which will help in the next steps. The text box is still very limited, and I'll be working on it in the future.

The current state of the Pygame UI implementation.

Objectives

Here's list of a few objectives I had in mind when starting this project. It contains some things I have completed and others that I still want to complete.

• Understand decorators.

  Although I understand how they work and how to implement them, I haven't yet found much use. Yet.

• Understand context managers.

  Not only have I understood how they work, I've developed the main.ReportManager class specifically to deal with creating model reports, something I already used to do in a more manual way before.

• Switch to Google's Style.

  Working on it!

• (WIP) Write a complete documentation with Sphinx.

  I have already worked with Sphinx in the past and personally loved it. This is a permanent work in progress, of course, but I'm currently testing a new theme (Furo) and haven't yet written a docstring for everything so it's particularly empty as of know. You can find the documentation here.

• [(WIP)] Implement an interface for real-time drawing and prediction.

  Development has started, using the Pygame module.

• Try to use PyTorch Lightning and PyTorch/XLA for accelerating training using cloud multi-core TPUs (in Google Colab).

  Despite knowing how to use Google Colab's GPUs for accelerating PyTorch code, TPUs and specifically multi-core parallelism is something I don't (yet) know how to work with.

• Perhaps learn and use Optuna for selecting training and Neural Networks hyperparameters.

• Develop more CNNs for testing.

• Finish developing functions for evaluating trained model's performances on the HASYv2 dataset.

  Using the same parameters as the ones used in the article.