​Нейронная сеть для создания плана квартиры

Французский разработчик Станислас Шайю в рамках работы над магистерской диссертацией представил систему, которая умеет генерировать планы помещений и заставлять их мебелью. Она базируется на алгоритме pix2pix и представляет собой конвейер, состоящий их трех моделей.

Первая модель генерирует вертикальную проекцию здания, используя данные о сооружениях Бостона из геоинформационной системы. Вторая модель требует вмешательства пользователя – ему необходимо нанести на эскиз окна и вход. После этого система рисует на плане различные помещения, которые делит цветом на типы, а также стены и проемы между ними. Третья модель, используя уже раскрашенный план, занимается расстановкой необходимой мебели по типу комнат.

Прочесть подробное описание алгоритма вы можете здесь, а протестировать его работу – здесь.

#hey_news
Источник: NVIDIA Developer Blog