Waste Classification CNN

PyTorch image classifier (recyclable vs. non-recyclable waste) comparing a custom CNN against ResNet18 transfer learning, trained and evaluated on AWS EC2 GPU infrastructure.

• Prepared and processed a multi-category image dataset; improved performance through data augmentation, class balancing and iterative model development.