Проект, в котором был реализован ряд моделей для генерации изображений с покемонами. Список генеративных моделей:
- from scratch: DCGAN, VAE
- transfer learning: StyleGAN, RuDALL-E
Для открытия и запуска ipynb-ноутбуков рекомендуется использовать среду Google Colab.
Для того, чтобы модель получилось запустить, необходимо скачать веса выбранной предобученной модели (по данной ссылке лежит zip-архив с весами моделей (3.11 ГБ): https://drive.google.com/file/d/14CEe1DobrZ0lN7jBSeHnxMjvEdwKwRH2/view?usp=sharing), и загрузить её на свой Google Drive.
После открытия ipynb-ноутбука с выбранной моделью, в первой ячейке с кодом (или в ячейках ниже - в зависимости от модели) необходимо указать пути до загруженных на Google Drive весов модели. После этого нужно сменить среду выполнения на GPU и модель готова к запуску (можно нажать “выполнить всё” в той же вкладке “среда выполнения”) . В ходе генерации (для DCGAN и VAE - в ходе обучения) в последней ячейке результаты будут либо сразу выводиться, либо сохраняться на локальном диске google colab.
- Оглоблин Владислав Семёнович
- Кузнецов Ерофей Алексеевич
- Сусляков Семён Сергеевич
- Тамбашова София Сергеевна
- Панов Богдан Александрович
- Цыплаков Олег Сергеевич
Титова Ксения Максимовна