Portfolio Project

Shape Classifier Demo

Handwritten Shape Recognition

Data Science Machine Learning Python PyTorch AWS Docker

Demo

STAR Summary

Situation
I wanted a model that can tell what shape someone drew.
Task
Owned the end-to-end build, from implementation through the final deliverable.
Action
  • Used Google's QuickDraw sketches and built train/validation splits.
  • Trained a small ResNet18 classifier in PyTorch.
  • Deployed a CPU-only AWS Lambda endpoint so the browser can request predictions.
Result
  • High accuracy on a five-shape subset (results vary by split and drawing style): circle, triangle, square, hexagon, and octagon.
  • Predictions return quickly once the Lambda container is warm; cold starts can take longer.

Notes

The live demo runs CPU-only serverless inference on five QuickDraw shape classes, so confidence is scoped to that small drawing task.