Back to Work
PyTorch Playing Card Classifier
AI / ML

PyTorch Playing Card Classifier

Deep learning model that classifies images of playing cards using transfer learning with EfficientNet-B0.

2024
PyTorchEfficientNettimmtorchvisionpandasmatplotlib

About This Project

A deep learning image classification project that identifies 53 playing card classes (standard deck plus Joker) using transfer learning with EfficientNet-B0. Built from scratch with custom Dataset/DataLoader implementations, CrossEntropyLoss, and Adam optimizer with CUDA GPU acceleration. The project emphasizes foundational deep learning concepts including data processing, model configuration, and training loop implementation.

Deep LearningPyTorchComputer VisionTransfer Learning