AutoML Vision

AutoML Vision is a Streamlit-based tool that lets you upload an image dataset, train an image classifier, and export the trained model — all without writing any ML code!

Features

• Upload a ZIP file containing class-wise folders of images
• Train a custom image classifier using MobileNetV2 + transfer learning
• Visualize training & validation accuracy
• Save the trained model (.h5 file) for reuse or deployment

Folder Structure

AutoML-Vision/ ├── main.py ├── utils.py ├── requirements.txt └── README.md