Details

В данной работе представлен алгоритм векторизации планов квартир по растровому изображению, основанный на комбинации методов статистической обработки изображений и методов глубокого машинного обучения. Задачи, которые решались в ходе исследования: 1. Исследование актуальности проблемы векторизации планов на современном рынке. 2. Изучение существующих подходов к векторизации планов квартир, выявление их недостатков. 3. Выбор архитектуры и обучение моделей глубокого машинного обучения для распознавания стен, окон, дверей и других объектов. 4. Разработка конвейера обработки изображений, принимающего на вход изображение, и выдающего векторный формат плана на выходе. Все данные для обучения моделей были собраны с сайтов по недвижимости. Разработанный метод аугментации данных позволил повысить метрики обучения, а также обучить модели, устойчивые к теням на изображении. В работе приведены результаты обучения. Произведено сравнение различных архитектур нейронных сетей для данной задачи. В результате был разработан алгоритм, способный векторизовать план квартиры по растровому изображению. Алгоритм может применяться для обработки различных типов планов квартир. Метод, предложенный в этой работе, может быть полезен для трехмерной реконструкции, расчета стоимости недвижимости, предложений по проектированию и т. д.

This work introduces the approach based on the combination of statistical image processing methods in a row of machine learning techniques that allow training robust model for the different floor plan topologies. The research set the following goals: 1. Studying of the relevance of the problem of vectorization of plans in the modern market. 2. Studying of existing approaches to vectorization of apartment plans, identifying their shortcomings. 3. The choice of architecture and teaching models of deep machine learning to recognize walls, windows, doors and other objects. 4. Development of an image processing pipeline that accepts an image as an input and outputs a vector plan format. All data for model training was collected from real estate sites. The developed method of data augmentation allowed to increase learning metrics, as well as to train models that are resistant to shadows in the image. The work presents the results of training. Comparison of various architectures of neural networks for this task is presented. As a result, an algorithm was developed that could vectorize the apartment plan according to a raster image. The algorithm can be used to process various types of apartment plans. The method proposed in this work can be useful for three-dimensional reconstruction, calculating the value of real estate, design proposals, etc.