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

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

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

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

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

1. Для каждого класса фиксируется свой основной цвет (переменные S и T, нужно задавать вручную для каждого изображения).
2. Расстояние между двумя цветами вычисляется как среднеквадратичное между яркостями компонент (функция dist).
3. Штраф данного пикселя за принадлежность к классу равен расстоянию от него до цвета класса, умноженному на коэффициент alpha.
4. Штраф за отнесение соседних пикселей к разным классам равен расстоянию между их цветами, умноженному на коэффициент beta.
5. В заготовке после сегментации пиксели изображения подкрашиваются в соответствующий цвет для наглядности.
6. Заготовкой пользоваться необязательно, программировать можно как на Python, так и на C++ (но заготовки для C++ нет).
7. Адаптируйте код под количество цветов, большее двух (как в примере из файла landscape.jpg).