public static Bitmap DrawAverage(Bitmap bmp, Center[] centers, Bitplane labels)
{
Bitmap3 bmp3 = new Bitmap3(bmp);
Bitmap3 result = DrawAverage(bmp3, centers, labels);
//result.LABtoRGB();
return(result.GetBitmap());
}
public Bitplane(Bitplane bitplane)
{
this.Width = bitplane.Width;
this.Height = bitplane.Height;
for (int y = 0; y < this.Height; ++y)
{
for (int x = 0; x < this.Width; ++x)
{
SetPixel(x, y, bitplane.GetPixel(x, y));
}
}
}
private static Bitmap3 DrawEdges(Bitmap3 image, Center[] centers, Bitplane labels, Color edgeColor)
{
Bitplane edges = new Bitplane(image.Width, image.Height);
Bitmap3 newImage = new Bitmap3(image.Width, image.Height, image.NumCh);
for (int y = 0; y < image.Height; ++y)
{
for (int x = 0; x < image.Width; ++x)
{
if (y == 0 || x == 0)
{
newImage.Bitplane[2].SetPixel(x, y, (int)Math.Round(image.Bitplane[2].GetPixel(x, y)));
newImage.Bitplane[1].SetPixel(x, y, (int)Math.Round(image.Bitplane[1].GetPixel(x, y)));
newImage.Bitplane[0].SetPixel(x, y, (int)Math.Round(image.Bitplane[0].GetPixel(x, y)));
continue;
}
double p = labels.GetPixel(x, y);
double p1 = labels.GetPixel(x, y - 1);
double p2 = labels.GetPixel(x - 1, y - 1);
double p3 = labels.GetPixel(x - 1, y);
bool edge = false;
if (((p1 != p) && (edges.GetPixel(x, y - 1) != 1)) || ((p2 != p) && (edges.GetPixel(x - 1, y - 1) != 1)) || ((p3 != p) && (edges.GetPixel(x - 1, y) != 1)))
{
edge = true;
edges.SetPixel(x, y, 1.0);
}
if (edge)
{
newImage.Bitplane[2].SetPixel(x, y, edgeColor.R);
newImage.Bitplane[1].SetPixel(x, y, edgeColor.G);
newImage.Bitplane[0].SetPixel(x, y, edgeColor.B);
}
else
{
newImage.Bitplane[2].SetPixel(x, y, (int)Math.Round(image.Bitplane[2].GetPixel(x, y)));
newImage.Bitplane[1].SetPixel(x, y, (int)Math.Round(image.Bitplane[1].GetPixel(x, y)));
newImage.Bitplane[0].SetPixel(x, y, (int)Math.Round(image.Bitplane[0].GetPixel(x, y)));
}
}
}
return(newImage);
}
private static Center[] CalculateNewCenters(Bitmap3 image, Center[] centers, Bitplane labels)
{
Center[] newCenters = new Center[centers.Length];
for (int i = 0; i < centers.Length; i++)
{
newCenters[i] = new Center(0, 0, 0, 0, 0, 0);
}
for (int y = 0; y < image.Height; ++y)
{
for (int x = 0; x < image.Width; ++x)
{
int centerIndex = (int)Math.Floor(labels.GetPixel(x, y));
if (centerIndex != -1)
{
double L = image.Bitplane[2].GetPixel(x, y);
double A = image.Bitplane[1].GetPixel(x, y);
double B = image.Bitplane[0].GetPixel(x, y);
newCenters[centerIndex].X += x;
newCenters[centerIndex].Y += y;
newCenters[centerIndex].L += L;
newCenters[centerIndex].A += A;
newCenters[centerIndex].B += B;
newCenters[centerIndex].COUNT += 1;
}
}
}
// Normalize
foreach (Center center in newCenters)
{
if (center.COUNT != 0)
{
center.X = Math.Round(center.X / center.COUNT);
center.Y = Math.Round(center.Y / center.COUNT);
center.L /= center.COUNT;
center.A /= center.COUNT;
center.B /= center.COUNT;
}
}
return(newCenters);
}
private static Bitmap3 DrawAverage(Bitmap3 image, Center[] centers, Bitplane labels)
{
Bitmap3 newImage = new Bitmap3(image.Width, image.Height, image.NumCh);
for (int y = 0; y < image.Height; ++y)
{
for (int x = 0; x < image.Width; ++x)
{
if (labels.GetPixel(x, y) != -1)
{
newImage.Bitplane[2].SetPixel(x, y, centers[(int)Math.Floor(labels.GetPixel(x, y))].L);
newImage.Bitplane[1].SetPixel(x, y, centers[(int)Math.Floor(labels.GetPixel(x, y))].A);
newImage.Bitplane[0].SetPixel(x, y, centers[(int)Math.Floor(labels.GetPixel(x, y))].B);
}
}
}
return(newImage);
}
/// <summary>
/// slic计算
/// </summary>
/// <param name="bmp"></param>
/// <param name="numberOfCenters"></param>
/// <param name="m"></param>
/// <param name="edgeColor"></param>
/// <returns></returns>
public static SlicPackage Run(Bitmap bmp, double numberOfCenters, double m, Color edgeColor)
{
Bitmap3 image = new Bitmap3(bmp);
Bitmap[] processedImages = new Bitmap[2];
// Convert RGB TO LAB
image.RGBtoLAB();
// Create centers
double S = Math.Sqrt((image.Width * image.Height) / numberOfCenters);
Center[] centers = CreateCenters(image, numberOfCenters, S);
Bitplane labels = new Bitplane(image.Width, image.Height);
labels.SetAllTo(-1);
for (int iteration = 0; iteration < 10; iteration++)
{
Bitplane lenghts = new Bitplane(image.Width, image.Height);
lenghts.Max();
int i = 0;
foreach (Center center in centers)
{
for (int k = (int)Math.Round(center.X - S); k < (int)Math.Round(center.X + S); k++)
{
for (int l = (int)Math.Round(center.Y - S); l < (int)Math.Round(center.Y + S); l++)
{
if (k >= 0 && k < image.Width && l >= 0 && l < image.Height)
{
double L = image.Bitplane[2].GetPixel(k, l);
double A = image.Bitplane[1].GetPixel(k, l);
double B = image.Bitplane[0].GetPixel(k, l);
double Dc = Math.Sqrt(Math.Pow(L - center.L, 2) + Math.Pow(A - center.A, 2) + Math.Pow(B - center.B, 2));
double Ds = Math.Sqrt(Math.Pow(l - center.Y, 2) + Math.Pow(k - center.X, 2));
double lenght = Math.Sqrt(Math.Pow(Dc, 2) + Math.Pow(Ds / 2, 2) * Math.Pow(m, 2));
if (lenght < lenghts.GetPixel(k, l))
{
lenghts.SetPixel(k, l, lenght);
labels.SetPixel(k, l, i);
}
}
}
}
i++;
}
centers = CalculateNewCenters(image, centers, labels);
}
//image.GetBitmap().Save(@"D:\Workspace\bmp\o.jpg");
//image = drawAverage(image, centers, labels);
//image.LABtoRGB();
//processedImages[0] = image.GetBitmap(); // Segmented
image = DrawEdges(image, centers, labels, edgeColor);
image.LABtoRGB();
processedImages[0] = image.GetBitmap(); // Segmented with Edge
image = DrawAverage(image, centers, labels);
image.LABtoRGB();
processedImages[1] = image.GetBitmap();
//center
string centerText = Newtonsoft.Json.JsonConvert.SerializeObject(centers);
//label
string labelText = Newtonsoft.Json.JsonConvert.SerializeObject(labels);
return(new SlicPackage()
{
Label = labelText,
Edge = processedImages[0],
Average = processedImages[1],
CENTER = centerText,
});
}
