Skip to content

CNN Image Classifier v1.0.0 - Initial Release

Latest

Choose a tag to compare

@Chetan9675 Chetan9675 released this 01 Jul 11:59
· 2 commits to main since this release
65a8fce

πŸŽ‰ 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 script
  • README.md - Comprehensive documentation
  • requirements.txt - All dependencies
  • LICENSE - 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