Какая GAN поддерживает альфа-канал?

Question

[luvjam]

Подскажите, пожалуйста, название ai, который поддерживает генерацию изображения с альфа-каналом, или вы знаете, как отредактировать готовый ai, чтобы получить 4-й канал. В настоящее время я обучаю styleGAN2-ada с 3 каналами, и это вызывает у меня много проблем.

Answer 1

[semyon_sempai]

Определённой архитектуры нет, но в чём проблема лапками переписать размерности, которые ожидает gan на вход? То есть по сути если на какой-то слой подаётся WxHx3, то поменяйте 3 на 4. Ну и сразу скажу, что jpg не поддерживают 4-х канальные изображения, так что работайте в png-формате. Так что берите любой gan, который вам понравится, и потратьте час своей жизни на правки в коде.