This project is a Python-based image classification program using ImageAI with a MobileNetV2 model. It demonstrates a simple machine learning workflow for classifying images, designed as a fun and educational project for practicing Python and ML concepts.
- Classifies images using a pre-trained MobileNetV2 model.
- Outputs the top predictions and their probabilities.
- Lightweight and easy to experiment with on custom images.
- Clone this repository.
- Ensure you have a compatible Python environment (see Disclaimer).
- Place an image in the project folder.
- Run the Python file:
python image_classification.py
View the predicted classes and probabilities.
Due to compatibility issues with ImageAI and the latest Python versions, this project must be run in a VM or environment using Python 3.8-3.11.