Nvidia представила нейросеть для создания реалистичных изображений

Компания NVIDIA представила нейросеть, способную превратить любой схематичный рисунок в фотореалистичное изображение за несколько секунд. Об этом сообщает TechCrunch.

Нейросеть GauGAN является демонстрацией возможностей нейросетевой платформы NVIDIA, но при этом представляет собой отдельное приложение.

GauGAN использует для обработки генеративно-состязательные нейросети (GAN): генератор создаёт изображение и передаёт его обученному на реальных фотографиях дискриминатору, который подсказывает генератору, как доработать изображение и что исправить. Для обучения Nvidia использовала 1 млн фотографий, полученных от Flickr.

При создании изображения GauGAN генерирует реалистичные детали, такие как отражение объектов в прудах, реках и на других водных поверхностях. Инструмент позволяет менять время суток и года: для этого достаточно переключиться с режима «трава» на «снег», и система автоматически поменяет окружение на картинке, например, уберёт листву с деревьев.

Пока GauGAN умеет работать только с пейзажами и элементами природы, но нейросеть можно обучить для создания и обработки изображений со зданиями, дорогами, людьми и другими объектами. GauGAN может стать инструментом, который поможет архитекторам, градостроителям, ландшафтным дизайнерам и разработчикам игр упростить создание прототипов и ускорить их редактирование, считают в Nvidia.

Компании ещё нужно немного доработать модель, считают в Nvidia. В скором времени GauGAN должна стать доступна на площадке для демонстрации возможностей нейросетей AI Playground, которую Nvidia также представила на GTC 2019.