Portfolio Project

Shape Classifier Demo

Handwritten Shape Recognition

Machine Learning Python PyTorch AWS Docker

Context

I wanted to create a model that recognizes handwritten shapes.

Approach

  • Downloaded images from Google's QuickDraw dataset to build training and validation splits.
  • Trained a compact ResNet18 using PyTorch Lightning.
  • Deployed a minimal AWS Lambda handler for serverless CPU inference from the browser.

Impact

  • Predicts circle, triangle, square, hexagon, or octagon from a single drawing with about 90% accuracy.
  • Demo shows responses return in under a second after a 10‑second warm‑up.

Links