This project is a deep learning model designed to classify different species of monkeys using images. Built with Convolutional Neural Networks (CNN) in Python using TensorFlow and Keras, it demonstrates a complete pipeline from data preprocessing to model evaluation and prediction.
- ๐ง Uses a custom CNN architecture for image classification
- ๐ Trained on a dataset of labeled monkey species images
- ๐งผ Includes image preprocessing and data augmentation (
ImageDataGenerator
) - ๐ฏ Evaluates model performance using accuracy and loss metrics
- ๐ Supports testing with new images for prediction
- Python 3
- TensorFlow / Keras
- NumPy, Matplotlib
- Jupyter Notebook
- Conv2D + ReLU + MaxPooling
- Dropout layers to reduce overfitting
- Flatten + Dense layers
- Softmax output layer for multi-class prediction
After training, the model achieves high accuracy on both training and validation sets. You can test the model by running a single image through it:
model.predict(processed_image)
- Clone the repo
- Make sure your dataset is correctly structured
- Run the notebook monkeys.ipynb step by step
- (Optional) Save and reload the model with model.save() and load_model()
๐ฎ Future Improvements
- Use transfer learning (e.g. with MobileNet or ResNet)
- Expand dataset for more robustness
- Add web interface for live predictions
๐ฉโ๐ป Author
Sara Sรกnchez Garcรญa LinkedIn | sara.sanchez.g29@gmail.com