Графические модели (курс лекций)/2014/Задание 3

Материал из MachineLearning.

Перейти к: навигация, поиск


Задача склеивания изображений: построение одного изображения по набору исходных.
Задача склеивания изображений: построение одного изображения по набору исходных.

Начало выполнения задания: 1 апреля 2014 г.;
Срок сдачи: 14 апреля 2014 г. (понедельник), 23:59.

Среда для выполнения задания — MATLAB. Неэффективная реализация кода может негативно отразиться на оценке.

Задание состоит из двух вариантов. Распределение вариантов задания по студентам:

Вариант 1: α-расширение Вариант 2: αβ-замена
Харациди О. Рыжков А.
Ульянов Д. Горелов А.
Антипов А. Львов С.
Сокурский Ю. Новиков А.
Алешин И. Шадриков А.
Ломов Н. Никифоров А.
Найдин О. Шабашев Ф.
Зиннурова Э. Подоприхин Д.
Корольков М. Арбузова Д.
Новиков М. (420) Петров Г.
Шахуро В. (420) Грингауз А. (320)
Ибадов Т. (420) Старцев М. (420)

Марковское случайное поле

Система соседства — прямоугольная решетка
Система соседства — прямоугольная решетка

Марковское случайное поле (MRF) — графическая модель, энергия которой записывается в виде:
 
E(X) = \sum_{i \in \mathcal{V}} \theta_i(x_i) + \sum_{(i, j) \in \mathcal{E}} \theta_{ij}(x_i, x_j), \quad x_i \in \mathcal{P},
где \mathcal{V} — множество индексов переменных, \mathcal{E} — система соседства, \theta_i: \mathcal{P} \to \mathbb{R} — унарные потенциалы, \theta_{ij}: \mathcal{P}\times\mathcal{P} \to \mathbb{R} — парные потенциалы. Обратите внимание, что в сумме по рёбрам (i, j) \in \mathcal{E} каждое ребро графа учитывается только один раз.

Рассмотрим модель со следующими ограничениями:

  • переменные  x_p дискретны и принимают значения из множества \mathcal{P} = \{1,\ldots,K\}, \;K \geq 2,
  • система соседства \mathcal{E} — прямоугольная решетка,

В рамках данного задания каждый студент должен реализовать один алгоритм минимизации энергии: вариант 1 реализует алгоритм α-расширение, вариант 2 — αβ-замена. Далее в задании алгоритм соответствующего варианта обозначается как алгоритм.

MRF для склеивания изображений

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

Для задачи склеивания энергия строится следующим образом:

  • Переменные x_p соответствуют пикселям финального изображения.
  • Значение каждой переменной соответствует номеру изображения исходного набора, из которого взят цвет соответствующего пикселя.
  • Используется стандартная 4-х связная система соседства.
  • Унарные потенциалы должны показывать, из каких изображений должны быть взяты некоторые пиксели (так называемые семена).
  • Парные потенциалы должны поощрять 1) короткие разрезы и 2) расположение разреза там, где изображения хорошо соответствуют друг другу.

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

Задание

  1. Вывести все формулы, использующиеся в вашей реализации алгоритма (сведение шага алгоритма к разрезу графа).
  2. Реализовать алгоритм, используя выданный код разрезов графов.
  3. Протестировать алгоритм на модельных данных.
  4. Реализовать процедуру решения задачи склеивания двух изображений. Построить не менее 1 хорошей композиции, состоящей из двух частей.
  5. Реализовать процедуру склеивания произвольного числа изображений. Построить не менее 1 хорошей композиции, состоящей из не менее чем 4-х частей.
  6. На задаче склеивания Вашего набора изображений сравнить работу алгоритмов α-расширение и αβ-замена. Реализацию недостающего алгоритма можно взять у товарища, выполняющего другой вариант (в отчете обязательно указывать, чей код вы используете). Требуется провести сравнение по энергии получаемого решения, по времени работы, по визуальному качеству решения. Все выводы должны быть подтверждены числами, графиками, картинками.
  7. Написать отчет в формате PDF с описанием всех проведенных исследований.

Обратите внимание, что для выполнения пунктов 4 и 5 каждый студент должен использовать уникальные (отличающиеся от изображений других студентов) изображения. Использование фотографий, сделанных собственноручно, будет поощряться. Использование картинок из интернета также допускается.

Композиция является хорошей, если границы на ней не более заметны, чем на композиции, приведённой в начале этого задания. Для достижения хорошего качества рекомендуется использовать редакторы изображений для выравнивания геометрии и цветов исходных изображений.


Спецификация реализуемых функций

Алгоритм
[labels, energy, time] = alphaExpansionGridPotts(unary, vertC, horC, metric)
[labels, energy, time] = alphaExpansionGridPotts(unary, vertC, horC, metric, options)
[labels, energy, time] = alphaBetaSwapGridPotts(unary, vertC, horC, metric)
[labels, energy, time] = alphaBetaSwapGridPotts(unary, vertC, horC, metric, options)
ВХОД
unary — унарные потенциалы, массив типа double размера N x M x K, где N — высота решетки, M — ширина решетки, K — количество меток.
vertC — коэффициенты  c_{pq}, соответствующие вертикальным ребрам, массив типа double размера (N — 1) x M;
horC — коэффициенты  c_{pq}, соответствующие горизонтальным ребрам, массив типа double размера N x (M — 1);
metric — расстояние между метками соседних переменных, массив типа double размера K x K;
options — (необязательный аргумент) набор дополнительных параметров, структура с полями
  'maxIter' — максимально допустимое число итераций алгоритма (по умолчанию = 500);
  'display' — параметр типа logical: если true, то при каждом запуске алгоритма разреза графа нужно выводить на экран номер итерации, номера обрабатываемых меток, текущее значение энергии;
  'numStart' — количество запусков из разных начальных приближений;
  'randOrder' — параметр типа logical: если true, то при каждом запуске использовать случайный порядок меток α и β;
ВЫХОД
labels — разметка, обладающая наименьшей энергией, массив типа double размера N x M;
energy — значения энергии на каждой итерации, вектор типа double длины, равной количеству итераций алгоритма;
time — время, пройденное с начала работы алгоритма до каждой итерации, вектор типа double длины, равной количеству итераций алгоритма;

Обратите внимание: в процедурах alphaExpansionGridPotts и alphaBetaSwapGridPotts параметры N, M, и K определяются неявно по размеру соответствующих элементов.

Склеивание
[resultImage, resultMask] = stichImages(images, seeds)
ВХОД
images — набор исходных изображений, cell array размера K x 1, где K – кол-во изображений. Все изображения должны быть одинакового разрешения и содержать ровно 3 цветовых канала.
seeds — маски, заданные пользователем, cell array размера K x 1. Каждый элемент - логический массив размера, равного разрешению изображения, в котором значение true означает, что соответствующий пиксель должен быть взят из соответствующего изображения.
ВЫХОД
resultImage — построенное изображение.
resultMask — маска ответа, массив типа double, по размеру равный разрешению изображения. Элемент results(i,j) равен k, если в построенном изображении пиксель (i, j) взят из исходного изображения номер k.

Рекомендации по выполнению задания

  1. Обратите внимание на область применимости алгоритма.
  2. При тестировании алгоритма необходимо следить за следующим:
    • после каждого применения разреза графа общая энергия не возрастает;
    • значение энергии, выдаваемое функцией graphCutMex, совпадает со значением энергии, подсчитанным независимой процедурой.
  3. Обратите внимание, что для достижения хорошего качества решения задачи склеивания, возможно придется изменить интерфейс, выданный в задании. При использовании измененного прототипа ОБЯЗАТЕЛЬНО нужно прислать и функцию, согласованную с выданным прототипом.

Данные для выполнения задания

graphCut — MATLAB интерфейс к разрезам графов.

Оформление задания

Выполненный вариант задания необходимо прислать письмом по адресу bayesml@gmail.com с темой «[ГМ14] задание 3, вариант X, фамилия». Убедительная просьба присылать выполненное задание только один раз с окончательным вариантом. Новые версии будут рассматриваться только в самом крайнем случае. Также убедительная просьба строго придерживаться заданной выше спецификации реализуемых функций. Очень трудно проверять большое количество заданий, если у каждого будет свой формат реализации.

Письмо должно содержать:

  • PDF-файл с описанием проведенных исследований (отчет должен включать в себя описание выполнения каждого пункта задания с приведением соответствующих графиков, изображений, чисел)
  • Все исходные файлы, реализованные в рамках настоящего задания. Убедитесь, что Ваши файлы работают и соответствуют прототипам!
  • Не менее двух склеенных изображений и соответствующих им наборов исходных.
Личные инструменты