Генерация реалистичных изображений нефтегазовой инфраструктуры на космических снимках с использованием диффузионных моделей

Полный текст

Введение Использование космических снимков позволяет решать многие задачи. Поэтому разработка моделей и методов обработки изображений является чрезвычайно актуальной. Цель исследования - оценка использования нейронных сетей для визуализации прокладки трубопроводов. Для решения поставленной нами задачи изучены возможности модели, предложенной авторами статьи “Tackling Few-Shot Segmentation in Remote Sensing via Inpainting Diffusion Model” [1]. Применение данной модели для прокладки трубопровода показало ее недостаточную эффективность. Для улучшения результатов использования модели провели дообучение на основе сформированного обучающего набора. Актуальность подхода связана с применением современных нейронных сетей и решаемой проблемы визуализации нефтегазовой инфраструктуры на космических снимках. 1. Методы 1.1. Архитектура нейронной сети Проанализируем принцип работы модели, предложенной в [1] и состоящей из нижеперечисленных «блоков». Stable Diffusion - это генеративная диффузионная модель, предназначенная для создания высококачественных изображений из текстовых описаний или на основе других изображений. Модель работает с пространством скрытых признаков, за счет этого способна достигать высокой эффективности при работе с изображениями. Обучение модели происходит на задаче «обратного шума» - она учится постепенно восстанавливать изображение из шума. При генерации Stable Diffusion «двигается в обратном порядке» от случайного шума к содержательному изображению [2; 3]. Основные компоненты Stable Diffusion: 1. Автокодировщик (VAE - Variational Autoencoder) - используется для перевода изо-бражений в пространство признаков и обратно и состоит из кодировщика (encoder), который сжимает изображение до латентного представления и декодировщика (decoder), восстанавливающего изображение из полученного кодировщиком представления [4]. Использование автокодировщика позволяет значительно ускорить обучение и генерацию по сравнению с работой с обычными пиксельными изображениями за счет снижения размерности данных и извлечения признаков. 2. UNet - это основной модуль, обрабатывающий скрытое пространство признаков. Архитектура UNet: симметричная сеть с блоками downsampling (понижающей дискретизации) и upsampling (повышающей дискретизации) [5]. В каждом блоке используются следующие составляющие: остаточные связи (residual), внимание (attention) и нормализация (group normalization). На вход модуля поступает зашумлённое пространство признаков, время шага в процессе диффузии (timestep) и текстовая подсказка (через cross-attention). 3. Text Encoder (CLIP Text Encoder) - разработанный OpenAI текстовый кодировщик CLIP (Contrastive Language - Image Pretraining) (ViT-B/32), преобразующий текстовую подсказку в вектор признаков. Эти признаки используются в UNet через cross-attention механизмы, чтобы направлять генерацию изображения [6]. 4. Scheduler (DDIM/DDPM) - управляет процессом добавления и удаления шума. Во время генерации работает в режиме «обратного процесса»: начиная со случайного шума, модель постепенно приближает его к изображению. Принцип работы этого метода заключается в следующем: 1. Текст → Вектор признаков: текст обрабатывается кодировщиком CLIP. 2. Латентное пространство: генерация на-чинается со случайного шума в латентном про-странстве. 3. Процесс диффузии: UNet итеративно очищает шум, направляясь вектором признаков текста. 4. VAE Decoder: финальное латентное пред-ставление декодируется обратно в изображение. RemoteCLIP - использовалась для обработки спутниковых и аэрофотоснимков (дистанционного зондирования) [7] и представляет собой адаптированную версию CLIP [6]. Модель обучается сопоставлять спутниковые изображения с текстовыми описаниями. Изображения и тексты кодируются в общее признаковое пространство, где близкие по смыслу пары находятся рядом. RemoteCLIP содержит следующие основные компоненты: 1. Кодировщик изображения (image en-coder) обычно используют ResNet [8] или Vision Transformer (ViT) [9]. Кодировщик преобразует изображение в векторные представления, с целью уменьшения объема данных для передачи используется квантование или компрессия. 2. Текстовый кодировщик (Text Encoder) (на сервере / в облаке) - использует трансформер (Transformer) для получения вектора признаков текстовой подсказки. 3. Коммуникационный интерфейс передает векторы признаков от клиента к серверу. Важными задачами являются сжатие, безопасность и стабильность передачи. 4. Fusion Module (на сервере) - сопоставляет визуальный и текстовый вектор признаков, вычисляя косинусное сходство. Взаимодействие: Stable Diffusion + RemoteCLIP. На первом этапе Stable Diffusion используется в режиме дорисовки (inpainting) для заполнения области изображения (с заданной маской) синтетическим объектом. Вместо текстового запроса подается визуальный образец (изображение с примером объекта нужного класса). На следующем этапе RemoteCLIP применяется в качестве кодировщика изображения, чтобы получить вектор признаков объекта из референсного изображения. Полученные векторы признаков передаются в Stable Diffusion как условие генерации, т.е. модель понимает, какой объект необходимо «нарисовать» в области маски. 1.2. Формирование обучающего набора По маршрутам существующих трубопроводов в QGIS (свободная бесплатная географическая информационная система с открытым кодом) были построены векторные слои. Затем для формирования обучающей выборки для маршрутов трубопроводов были получены космические снимки и векторные данные с точным расположением труб, которые использовались для дообучения модели. Формирование обучающего набора начинается с подготовки слоев в QGIS и преобразования координат. Загружается векторный линейный слой (маршрут трубопровода) и растровая подложка, после чего определяется необходимость перехода между системами координат. Если требуется, то все линейные объекты из слоя преобразуются к системе координат подложки. Далее выполняется создание фрагментов карты (тайлов) - квадратных зон с центрированием вдоль линейного объекта. Количество и расположение рассчитывается с учетом процента перекрытия соседних тайлов. Каждый полученный участок добавляется в память в виде полигона с уникальным идентификатором. После этого для каждого полученного полигона проверяется, пересекает ли его хотя бы одна из трубопроводных линий. Если пересечение обнаружено, вычисляются прямоугольные границы (bbox) в пиксельных координатах относительно тайла. Координаты полигонов bbox сохраняются в текстовых файлах в формате YOLO для дальнейшего использования в задаче обнаружения объектов. На этапе сохранения для каждого фрагмента карты формируется временный файл GeoTIFF в пределах границ тайла, который затем преобразуется в формат JPEG с помощью инструментов GDAL с последующим удалением исходного файла. Файл с текстовыми координатами bbox сохраняется под тем же названием, что и изображение. Полученный набор для дообучения имеет следующую рекомендованную структуру [10], представленную на рис. 1. Рис. 1. Структура обучающего набора данных: bbox - вектор объекта; images - соответствующее изображение И с т о ч н и к: выполнено Ш.М. Гаджиевым Figure 1. Training dataset structure: bbox - object bounding box coordinates; images - corresponding image S o u r c e: by Sh.M. Gadzhiev 2. Результаты и обсуждение 2.1. Дообучение Дообучение требовалось провести из-за отсутствия в начальном обучающем наборе класса с трубами. Следовательно, модель была не способна их генерировать. Для генерации необходимы изначальное изображение, маска области и непосредственно объект, который мы будем вставлять. При подаче в качестве референса участка трубопровода созданная труба выглядела как белая линия (рис. 2, a, б). Во время дообучения можно заметить, как с каждой эпохой модель справлялась с задачей генерации труб лучше и лучше. В качестве метрики была выбрана сглаженная ошибка валидации (экспоненциальное среднее). Приведем сравнение величины на 5-й и 9-й итерации, где значение уменьшилось с 0,86 до 0,77 (рис. 3 и 4). После дообучения модели на основе обучающей выборки с учетом вектора координат трубы на новый класс - «трубы» модель стала лучше встраивать трубопровод в космический снимок. Результат представлен на рис. 5. a б Рис. 2. Применение оригинальной модели для генерации трубопровода: a - подаваемые параметры; б - полученный результат И с т о ч н и к: выполнено Ш.М. Гаджиевым Figure 2. Application of the original model for pipeline generation: a - input parameters; б - obtained result S o u r c e: by Sh.M. Gadzhiev Рис. 3. Значение потерь на 5-й эпохе И с т о ч н и к: выполнено Ш.М. Гаджиевым Figure 3. Loss value at epoch 5 S o u r c e: by Sh.M. Gadzhiev Рис. 4. Значение потерь на 9-й эпохе И с т о ч н и к: выполнено Ш.М. Гаджиевым Figure 4. Loss value at epoch 9 S o u r c e: by Sh.M. Gadzhiev a б Рис. 5. Применение дообученной модели для генерации трубопровода: a - подаваемые параметры; б - полученный результат И с т о ч н и к: выполнено Ш.М. Гаджиевым Figure 5. Application of the fine-tuned model for pipeline generation: a - input parameters; б - obtained result S o u r c e: by Sh.M. Gadzhiev Заключение В исследовании рассматривалась проблема визуализации объектов нефтегазовой индустрии на снимках дистанционного зондирования Земли. В ходе работы был изучен и доработан подход на основе диффузионных моделей. Для улучшения результатов изучаемой модели было осуществлено дообучение на основе сформированного набора данных. После нашего дообучения модели повысилось качество снимков, полученных в ходе генерации реалистичных изображений нефтегазовой инфраструктуры на космических снимках по сравнению с моделью, в референсной статье.

Об авторах

Василий Константинович Лобанов

Российский университет дружбы народов

Email: lobanov_vk@pfur.ru
ORCID iD: 0000-0001-8163-9663
SPIN-код: 7266-5340

старший преподаватель кафедры механики и процессов управления, инженерная академия

Российская Федерация, 117198, г. Москва, ул. Миклухо-Маклая, д. 6

Мария Сергеевна Кондрашина

Российский университет дружбы народов

Автор, ответственный за переписку.
Email: 1132236536@rudn.ru
ORCID iD: 0009-0008-8526-9143

магистрант кафедры механики и процессов управления, инженерная академия

Российская Федерация, 117198, г. Москва, ул. Миклухо-Маклая, д. 6

Шамиль Магомедэминович Гаджиев

Российский университет дружбы народов

Email: 1132236511@rudn.ru
ORCID iD: 0009-0006-1570-4133

магистрант кафедры механики и процессов управления, инженерная академия

Российская Федерация, 117198, г. Москва, ул. Миклухо-Маклая, д. 6

Максад Ширинбекович Сокибеков

Российский университет дружбы народов

Email: 1032185455@rudn.ru
ORCID iD: 0009-0009-0261-7374

магистрант кафедры архитектура, реставрация и дизайн, инженерная академия

Российская Федерация, 117198, г. Москва, ул. Миклухо-Маклая, д. 6

Список литературы

Дополнительные файлы