Своя модель классификации (кошки / собаки / мужчины / женщины)#

Дообучение MobileNetV2 на картинках из папки data/ (одна подпапка = один класс). Веса MobileNetV2 (ImageNet) скачиваются автоматически при первом запуске.

uv sync
uv run python main.py      # обучение -> model.keras
uv run python predict.py   # предсказание для картинки из tests/

Можно добавить свои классы: создайте новую папку в data/ и положите туда 5+ картинок.