My project from YSDA Deep Generative Models course
Проект делался на dataspehere (обучение моделей, хранение датасета). Тесты и инференс производились на локальном компьютере.
Информация по структуре:
- train_main.ipynb - ноутбук с обучением модельки на датасфере
- first_look_and_save.ipynb - ноутбук с инициализацией датасета на датасфере и первым взглядом на датасет
- data_analysis.ipynb - EDA
- clustering.ipynb - кластеризация на три класса (blond/black/brown)
- explore_results_inference.ipynb - инференс финальной модельки
- pdf с презентацией проекта
- src - основные модули (нужны для инференса)
- samples - картинки во время обучения, me - картинка меня, partition1 - кусок датасета, нужен для инференса
Остальные файлы можно найти на яндекс диске:
- Два чекпоинта обученных моделей: https://disk.yandex.ru/d/8l_Vk_Bc5W4nDg
- Датасет и аннотации: https://disk.yandex.ru/d/j78hZ7zDkbvNhg