Fashion Image Classification

Код обучения модели классификации изображений одежды для проекта Smart Wardrobe.

Описание

Проект содержит скрипт для подготовки датасета, обучения модели EfficientNetV2B0 и экспорта результата в TFLite float16.

Скрипт рассчитан на запуск в Google Colab.

Что делает скрипт

• Загружает датасет fashion-product-images-small с Kaggle
• Распределяет изображения по классам одежды
• Создаёт train/validation split
• Обучает модель EfficientNetV2B0
• Выполняет fine-tuning модели
• Строит classification report и confusion matrix
• Экспортирует модель в .keras
• Конвертирует модель в .tflite для Android-приложения
• Сохраняет файл с названиями классов

Классы

• tshirt
• shirt
• sweater
• jacket
• dress
• jeans
• trousers
• shorts
• skirt
• shoes
• bag
• accessory

Стек технологий

• Python
• Google Colab
• TensorFlow / Keras
• EfficientNetV2B0
• TensorFlow Lite
• Pandas
• NumPy
• scikit-learn
• Matplotlib
• Kaggle API

Как запустить

1. Открыть скрипт в Google Colab.
2. Загрузить свой файл kaggle.json.
3. Запустить ячейку со скриптом.
4. После выполнения модель и дополнительные файлы будут сохранены в Google Drive.

Приватные файлы

В репозиторий не добавляются:

• kaggle.json
• скачанный датасет
• подготовленная папка с изображениями
• .keras модель
• .tflite модель
• checkpoints
• training logs

Статус проекта

Учебный ML-проект, подготовленный для интеграции модели классификации одежды в Android-приложение.