Внедрение BEGAN (граничное равновесие GAN) с использованием CNTK Python API

Я нашел реализацию для BEGAN с использованием CNTK. (https://github.com/2wins/BEGAN-cntk)

Здесь используется набор данных MNIST вместо Celeb A, который использовался в исходной статье. Однако я не понимаю результирующие изображения, которые выглядят вполне детерминированными: Выходные изображения обученного генератора (итер: 30000)

Я ожидаю, что для разных сэмплов шума будут получены разные результаты. Но он не делает этого независимо от каких-либо гиперпараметров. Какая часть кода создает проблему?

Пожалуйста, объясните это.


person David    schedule 04.05.2017    source источник
comment
BEGAN-cntk работает хорошо через 1 год. Возможно, проблема была в CNTK.   -  person David    schedule 03.06.2018


Ответы (2)


Используйте более высокую гамму (например, гамма = 1 или 1,3, на самом деле больше 1). Тогда он, конечно, улучшится, но не сделает его совершенным. Сделайте достаточное количество итераций, например, 200 000.

Пожалуйста, внимательно посмотрите на бумагу. Там сказано, что параметр гамма управляет разнообразием.

Один из полученных мною результатов: введите здесь описание изображения.

Я также ищу лучшие параметры и лучшие результаты, но пока не нашел.

person jachilles    schedule 05.06.2017
comment
Приведенные выше изображения получены на CNTK? К сожалению, ваш совет не влияет на результат. - person David; 09.06.2017
comment
@David Нет. Я использовал BEGAN от Heumi (github). - person jachilles; 10.06.2017

Похоже, ваша модель может зависнуть в определенном режиме. Одной из идей было бы добавить дополнительное условие к меткам классов. Для преодоления таких ограничений были предложены условные GAN.

http://www.foldl.me/uploads/2015/conditional-gans-face-generation/paper.pdf

Это идея, которую стоит изучить.

person Sayan Pathak    schedule 20.05.2017
comment
Я не понимаю, что программное обеспечение, используемое в реализации, изменилось, и результаты изменились. Оригинальная сетевая модель для CelebA также не работает для набора данных, который представлен в статье. - person David; 09.06.2017