Нейросеть научили расставлять мебель в комнате
Выпускник Федеральной политехнической школы Лозанны (EPFL) и Гарвардской высшей школы дизайна (GSD) Станислас Шайю разработал компьютерную систему на основе алгоритма междоменного преобразования объектов, которая способна создавать планы помещений. По сути она представляет собой набор из трёх моделей pix2pix, обученных на различных данных. Первая модель создаёт вертикальную проекцию помещения.
Она была обучена на массиве данных о зданиях из геоинформационной системы. Набор для обучения состоял из пар изображений, на одном из которых были контуры земельного участка, а на втором – контуры здания. Следующая модель использует полученную проекцию помещения и просит пользователя обозначить на ней двери и окна. После этого нейросеть делит помещение на комнаты разного назначения, окрашивая их в разные цвета.
Эта модель была обучена более чем на 800 планах квартир. Третья модель использует полученный план квартиры и наносит на него мебель, соответствующую каждой из комнат. Она была обучена на парах исходных и конечных изображений квартир. Шайю опубликовал алгоритм в открытом доступе, чтобы испытать его мог любой желающий – правда, качество получаемых изображений пока оставляет желать лучшего.