π CNN Image Classifier v1.0.0
π What's New
This is the first stable release of the CNN Image Classifier project!
β¨ Features
- Complete CNN Implementation: 4-layer convolutional neural network with batch normalization
- CIFAR-10 Support: Pre-configured for CIFAR-10 dataset classification
- Advanced Training: Early stopping, learning rate scheduling, and model checkpointing
- Data Augmentation: Random rotation, flipping, and cropping
- GPU Support: Automatic CUDA detection and utilization
- Comprehensive Evaluation: Confusion matrices, classification reports, and visualizations
- Progress Tracking: Real-time training progress with tqdm
π― Performance
- Accuracy: 80-85% on CIFAR-10 test set
- Training Time: 10-15 minutes on GPU, 30-60 minutes on CPU
- Model Size: ~2MB
- Parameters: ~1.2M trainable parameters
π¦ What's Included
pytorch_classifier.py- Main training scriptREADME.md- Comprehensive documentationrequirements.txt- All dependenciesLICENSE- MIT License
π Quick Start
# Clone the repository
git clone https://github.com/yourusername/cnn-image-classifier.git
cd cnn-image-classifier
# Install dependencies
pip install -r requirements.txt
# Run the classifier
python pytorch_classifier.py