Computer Vision & ML

Image Classification with PyTorch

CNN image classifier in PyTorch fine-tuning a pretrained ResNet-18 backbone on the CIFAR-10 dataset with data augmentation and learning-rate schedulers.

Key Architectural & System Features

Built and trained CNN classifier fine-tuning pretrained ResNet-18
Applied data augmentation and learning-rate scheduler
Evaluated performance using accuracy metrics and detailed confusion matrix

Technologies & Tools Stack

PyTorchCNNResNetCIFAR-10