Циклическая генеративная состязательная сеть с двойным специализированным механизмом внимания

LAO Jun-ming ,  

YE Wu-jian ,  

LIU Yi-jun ,  

YUAN Kai-yi ,  

摘要

Существующие методы генерации изображений на основе циклических генеративных состязательных сетей достигают хороших результатов в задачах преобразования несопоставимых изображений за счет внедрения независимых универсальных модулей внимания, однако при этом увеличивается сложность модели и время обучения, а также сложно учесть все детали ключевых областей изображения, оставляя пространство для улучшения качества генерации. Для решения указанных проблем предложена циклическая генеративная состязательная сеть с двойным специализированным механизмом внимания (Dual-SAG-CycleGAN), в которой генератор и дискриминатор направляются разными механизмами внимания. Во-первых, предложен специализированный модуль внимания под названием SAG (Special Attention-mechanism Guided) для руководства генератором, который повышает качество сгенерированных изображений и одновременно уменьшает сложность сети; затем для дискриминатора используется модуль с специализированным механизмом внимания на основе CAM (Class Activation Mapping), подавляющий нерелевантный шум, создаваемый генератором; наконец, предложена циклическая функция потерь согласованности с маской фона для направления генератора к созданию более точных масок, лучше поддерживающих преобразование изображений. Эксперименты показали, что предложенный метод по сравнению с существующими аналогами снижает количество параметров модели примерно на 32.8%, ускоряет обучение на 34.5%, при этом минимальные значения KID и FID достигают 1.13 и 57.54 соответственно, обеспечивая более высокое качество изображений.

关键词

Генеративные состязательные сети;Преобразование несопоставимых изображений;Специализированный механизм внимания;Циклическая функция потерь;Генерация изображений

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website