Digit Recognition

I created a number writing tool that you can check out here. Technologies Used Github Pages (Hosting & Deployment) React (UI) Python (Prototyping) Pytorch (Model design and training) Typescript (Model prediction) Vite (Building Process) Codebase The entire codebase is available on Github. Design For this web application, I utilized a React frontend and implemented a LeNet model using Pytorch. The model was trained on the MNIST dataset. Project Structure model_exploration: This section covers the training and model exploration processes. notebook.ipynb: A Jupyter notebook containing the model exploration. LeNet5.py: The Pytorch definition of the model. ui: This contains the React application that constitutes the frontend with a standard React structure. Model Training The Pytorch-based LeNet model code can be found here. The model consists of 5 blocks, with 3 being convolution blocks and 2 being feedforward blocks. The activation function used for all blocks except the final layer is Tanh. Training was conducted on the MNIST dataset for 10 epochs, resulting in an impressive test accuracy of 98.6%. ...

March 10, 2023 · 2 min · Abhay Vashist