A simple deep learning project using a Convolutional Neural Network (CNN) trained on the CIFAR-10 dataset. It classifies custom images (like cars, trucks, cats, etc.) using a pre-trained TensorFlow model.
- CIFAR-10: 60,000 images across 10 categories
- Classes: Plane, Car, Bird, Cat, Deer, Dog, Frog, Horse, Ship, Truck
- Python
- TensorFlow / Keras
- OpenCV
- NumPy
- Matplotlib
image-classifier-tensorflow/ ├── classifier.py # Main prediction script ├── Image Classification/ │ ├── image_classifier.keras # Pretrained model │ ├── car.jpg # Sample test image ├── README.md
- Loads CIFAR-10 dataset (optional for reference)
- Loads pre-trained
.kerasmodel - Reads and resizes a custom image (e.g.,
car.jpg) - Normalizes the image
- Runs prediction and displays the predicted class
git clone https://github.com/mahiyafatema/image-classifier-tensorflow.git
cd image-classifier-tensorflow
pip install tensorflow opencv-python matplotlib numpy
python classifier.py
### Make sure your .keras model file and image (car.jpg) are in the correct folder as specified in the script.
## Example Output
Prediction is Car
---
If you want, I can generate this as a markdown file for you to upload — or help you add a section like "Future Improvements" or "Contact Info." Would you like that?