Алгоритмы и структуры данных 2 2017/2018/Segmentation

Сегментация изображений

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

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

Заготовка с кодом

Примеры изображений

Пояснения по заданию и коду:

• Для каждого класса фиксируется свой основной цвет (переменные S и T, нужно задавать вручную для каждого изображения)
• Расстояние между двумя цветам вычисляется как среднеквадратичное между яркостями компонент (функция dist)
• Штраф данного пикселя за принадлежность классу равен расстоянию от него до цвета класса
• Штраф для отнесения соседних пикселей к разным классам равен расстоянию между их цветами
• В заготовке после сегментации пиксели изображения подкрашиваются в соответствующий цвет для наглядности
• Заготовкой пользоваться не обязательно, программировать можно как на Python, так и на C++ (но заготовки для C++ нет)