/Image-classification-VGG16

This project modifies the classic VGG16 architecture to classify images into four distinct categories with high accuracy. It incorporates data augmentation, dynamic learning rate adjustments, and comprehensive performance evaluation using accuracy metrics and confusion matrices. Built with PyTorch and supported by a suite of powerful libraries

Primary LanguagePython

Watchers