antonypotapenko/cyclegan-lab5

CycleGAN Day ↔ Night Demo

Данный Space представляет собой лёгкое Streamlit-приложение для демонстрации работы модели CycleGAN, обученной на задаче преобразования изображений между двумя доменами:

• —A → B: дневное изображение → ночное изображение
• —B → A: ночное изображение → дневное изображение

В приложении используются два сохранённых генератора, полученных в ноутбуке после обучения модели:

• —artifacts/gen_a2b_custom.pth
• —artifacts/gen_b2a_custom.pth

Возможности приложения

Приложение позволяет:

• —загружать изображение из дневного или ночного домена;
• —выполнять инференс на CPU;
• —получать результат преобразования в противоположный домен;
• —удобно демонстрировать качество обученной модели через веб-интерфейс.

Какие файлы нужно загрузить в репозиторий Space

Для корректной работы приложения в репозиторий Hugging Face Space необходимо добавить следующие файлы:

• —app.py
• —modeling.py
• —requirements.txt
• —Dockerfile
• —README.md
• —artifacts/gen_a2b_custom.pth
• —artifacts/gen_b2a_custom.pth

Необязательное ускорение на CPU

Для более быстрой загрузки модели и немного более быстрого инференса на CPU можно заранее локально экспортировать генераторы в формат TorchScript.

Для этого нужно выполнить команду:

python export_jit.py