Skip to content

Commit

Permalink
Formatted GlobalMatting class to standard format
Browse files Browse the repository at this point in the history
  • Loading branch information
@cs-util
cs-util committed Jul 3, 2024
1 parent 26850ff commit 7de4c6e
Showing 1 changed file with 28 additions and 28 deletions.
56 changes: 28 additions & 28 deletions CsCore/PlainNetClassLib/src/Plugins/CsCore/com/csutil/algorithms/images/GlobalMatting.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,7 @@ public void ExpansionOfKnownRegionsHelper(byte[] image, byte[] trimap, int r, fl
var im = GetColorAt(image, x, y);

for (var j = y - r; j <= y + r; ++j) {
for (var i = x-r; i <= x + r; ++i) {
for (var i = x - r; i <= x + r; ++i) {
if (i < 0 || i >= w || j < 0 || j >= h)
continue;
if (!ColorIsValue(trimap, i, j, bytesPerPixel, 0) && !ColorIsValue(trimap, i, j, bytesPerPixel, 255))
Expand All @@ -237,21 +237,20 @@ public void ExpansionOfKnownRegionsHelper(byte[] image, byte[] trimap, int r, fl
if (!(pd <= r) || !(cd <= c))
continue;
if (ColorIsValue(trimap, i, j, bytesPerPixel, 0))
SetColorAt(trimap, x, y, new byte[] {1,1,1,255});
SetColorAt(trimap, x, y, new byte[] { 1, 1, 1, 255 });
else if (ColorIsValue(trimap, i, j, bytesPerPixel, 255))
SetColorAt(trimap, x, y, new byte[] {254,254,254,255});
SetColorAt(trimap, x, y, new byte[] { 254, 254, 254, 255 });
}
}
}
}

for (int x = 0; x < width; ++x)
for (int y = 0; y < height; ++y)
{
for (int y = 0; y < height; ++y) {
if (ColorIsValue(trimap, x, y, bytesPerPixel, 1))
SetColorAt(trimap, x, y, new byte[] {0,0,0,255});
SetColorAt(trimap, x, y, new byte[] { 0, 0, 0, 255 });
else if (ColorIsValue(trimap, x, y, bytesPerPixel, 254))
SetColorAt(trimap, x, y, new byte[] {255,255,255,255});
SetColorAt(trimap, x, y, new byte[] { 255, 255, 255, 255 });

}
}
Expand Down Expand Up @@ -286,10 +285,10 @@ public void ExpansionOfKnownRegionsHelper(byte[] trimap, int r, float c) {
float cd = ColorDist(I, I2);

if (pd <= r && cd <= c) {
if(ColorIsValue(trimap, i, j, bytesPerPixel, 0))
SetColorAt(updatedTrimap, x, y, new byte[] {1,1,1,255});
if (ColorIsValue(trimap, i, j, bytesPerPixel, 0))
SetColorAt(updatedTrimap, x, y, new byte[] { 1, 1, 1, 255 });
else if (ColorIsValue(trimap, i, j, bytesPerPixel, 255))
SetColorAt(updatedTrimap, x, y, new byte[] {254,254,254,255});
SetColorAt(updatedTrimap, x, y, new byte[] { 254, 254, 254, 255 });
}
}
}
Expand All @@ -300,9 +299,10 @@ public void ExpansionOfKnownRegionsHelper(byte[] trimap, int r, float c) {
for (int x = 0; x < w; ++x) {
for (int y = 0; y < h; ++y) {
if (ColorIsValue(updatedTrimap, x, y, bytesPerPixel, 1))
SetColorAt(trimap, x, y, new byte[] {0,0,0,255});
SetColorAt(trimap, x, y, new byte[] { 0, 0, 0, 255 });
else if (ColorIsValue(updatedTrimap, x, y, bytesPerPixel, 254))
SetColorAt(trimap, x, y, new byte[] {255,255,255,255});;
SetColorAt(trimap, x, y, new byte[] { 255, 255, 255, 255 });
;
}
}
}
Expand All @@ -319,21 +319,21 @@ public void ErodeFB(ref byte[] trimap, int r) {
for (int y = 0; y < h; ++y) {
for (int x = 0; x < w; ++x) {
if (ColorIsValue(trimap, x, y, bytesPerPixel, 0))
SetColorAt(background, x, y, new byte[] {1,1,1,255});
SetColorAt(background, x, y, new byte[] { 1, 1, 1, 255 });
else if (ColorIsValue(trimap, x, y, bytesPerPixel, 255))
SetColorAt(foreground, x, y, new byte[] {1,1,1,255});
SetColorAt(foreground, x, y, new byte[] { 1, 1, 1, 255 });
}
}

// Erode the foreground and background
byte[] erodedBackground = Filter.Erode(background, w, h, 4,r);
byte[] erodedForeground = Filter.Erode(foreground, w, h, 4,r);
byte[] erodedBackground = Filter.Erode(background, w, h, 4, r);
byte[] erodedForeground = Filter.Erode(foreground, w, h, 4, r);

// Increase unknown region
for (int y = 0; y < h; ++y) {
for (int x = 0; x < w; ++x) {
if (ColorIsValue(erodedBackground, x, y, bytesPerPixel, 0) && ColorIsValue(erodedForeground, x, y, bytesPerPixel, 0))
SetColorAt(trimap, x, y, new byte[]{128,128,128,255}); // Set to unknown
SetColorAt(trimap, x, y, new byte[] { 128, 128, 128, 255 }); // Set to unknown
}
}
}
Expand All @@ -348,7 +348,7 @@ private bool ColorIsValue(byte[] array, int x, int y, int bytePerPixel, int valu
}
return result;
}

// Helper method to erode an image
private byte[] Erode(byte[] image, int w, int h, int r) {
byte[] erodedImage = new byte[image.Length];
Expand All @@ -363,7 +363,7 @@ private byte[] Erode(byte[] image, int w, int h, int r) {
int nx = x + dx;
int ny = y + dy;
if (nx >= 0 && nx < w && ny >= 0 && ny < h) {
int idx = (ny * w + nx)*bytesPerPixel;
int idx = (ny * w + nx) * bytesPerPixel;
// If any pixel in the neighborhood is 0, erode the current pixel
if (image[idx] == 0) {
erodePixel = true;
Expand All @@ -382,7 +382,7 @@ private byte[] Erode(byte[] image, int w, int h, int r) {

return erodedImage;
}

// Helper method to generate a random float between 0 and 1
private float RandomFloat() {
Random rand = new Random();
Expand Down Expand Up @@ -546,21 +546,21 @@ public void GlobalMattingHelper(byte[] trimap, out byte[] foreground, out byte[]
int idx = (y * width + x) * bytesPerPixel;
switch (trimap[idx]) {
case 0:
SetColorAt(alpha, x, y, new byte[]{0,0,0,255});
SetColorAt(alpha, x, y, new byte[] { 0, 0, 0, 255 });
conf[idx] = 255;
SetColorAt(foreground, x, y, new byte[] { 0, 0, 0 , 255});
SetColorAt(foreground, x, y, new byte[] { 0, 0, 0, 255 });
break;
case 128:
Sample s = samples[y][x];
var alphaValue = (byte)(255 * s.alpha);
SetColorAt(alpha, x, y, new byte[]{alphaValue, alphaValue, alphaValue, 255});
SetColorAt(alpha, x, y, new byte[] { alphaValue, alphaValue, alphaValue, 255 });
conf[idx] = (byte)(255 * Math.Exp(-s.cost / 6));
Point p = foregroundBoundary[s.fi];
byte[] color = GetColorAt(image, p.X, p.Y);
SetColorAt(foreground, x, y, color);
break;
case 255:
SetColorAt(alpha, x, y, new byte[]{255,255,255,255});
SetColorAt(alpha, x, y, new byte[] { 255, 255, 255, 255 });
conf[idx] = 255;
byte[] fgColor = GetColorAt(image, x, y);
SetColorAt(foreground, x, y, fgColor);
Expand Down Expand Up @@ -607,7 +607,7 @@ public void RunGlobalMatting(byte[] trimap, out byte[] foreground, out byte[] al
// Check if image and trimap have the same size
if (image.Length != trimap.Length)
throw new ArgumentException("image and trimap must have the same size");

// Call the helper function to perform the global matting
GlobalMattingHelper(trimap, out foreground, out alpha, out conf);
}
Expand All @@ -623,9 +623,9 @@ public byte[] RunGuidedFilter(byte[] alpha, int r, double eps) {
for (int x = 0; x < width; x++) {
for (int y = 0; y < height; y++) {
var col = GetColorAt(guidedIm, x, y);
var temp = new double[] { col[0], col[1], col[2]};
var temp = new double[] { col[0], col[1], col[2] };
var max = (byte)temp.Max();
SetColorAt(guidedIm, x, y, new []{max, max, max, (byte)255});
SetColorAt(guidedIm, x, y, new[] { max, max, max, (byte)255 });
}
}
return guidedIm;
Expand Down

0 comments on commit 7de4c6e

Please sign in to comment.