Вот код C # для 1- мерного квадратного евклидова преобразования расстояния в соответствии с работой Фельценшвальда и Хуттенлохера :
private static void DistanceTransform(double[] dataInput, ref double[] dataOutput)
{
int n = dataInput.Length;
int k = 0;
int[] v = new int[n];
double[] z = new double[n + 1];
v[0] = 0;
z[0] = Double.NegativeInfinity;
z[1] = Double.PositiveInfinity;
double s;
for (int q = 1; q < n; q++)
{
while (true)
{
s = (((dataInput[q] + q * q) - (dataInput[v[k]] + v[k] * v[k])) / (2.0 * q - 2.0 * v[k]));
if (s <= z[k])
{
k--;
}
else
{
break;
}
}
k++;
v[k] = q;
z[k] = s;
z[k + 1] = Double.PositiveInfinity;
}
k = 0;
for (int q = 0; q < n; q++)
{
while (z[k + 1] < q)
{
k++;
}
dataOutput[q] = ((q - v[k]) * (q - v[k]) + dataInput[v[k]]);
}
}
Это можно легко использовать для двоичных изображений и изображений в градациях серого, применив их сначала к столбцам изображений, а затем к строкам (или наоборот, конечно).
Преобразование действительно очень быстрое.
Вот исходные и выходные изображения:
Черные пиксели имеют значение 0, а белые имеют некоторое большое значение (должно быть больше, чем наибольшее возможное квадратное расстояние на изображениях, но не бесконечность), так что преобразование возвращает расстояние от черных пикселей, а белые пропускаются.
Чтобы получить истинное евклидово преобразование расстояния, просто возьмите квадратный корень каждого пикселя из выходного изображения.