alexnikko / stargan_v2_hair_recolor

My project from YSDA Deep Generative Models course

stargan_v2_hair_recolor

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