pretty high confidence, could increase crop on the right if face issues later.

Author: tjaycodes4you

edge_bg.png

@@ -190,6 +190,23 @@ func ExtractMask(img *image.RGBA) *image.Alpha {
 	return mask
 }
 
+func GetMaskOutline(mask *image.Alpha) *image.Alpha {
+	bounds := mask.Bounds()
+	outline := image.NewAlpha(bounds)
+	for y := bounds.Min.Y + 1; y < bounds.Max.Y-1; y++ {
+		for x := bounds.Min.X + 1; x < bounds.Max.X-1; x++ {
+			if mask.AlphaAt(x, y).A > 0 {
+				// Check neighbors
+				if mask.AlphaAt(x+1, y).A == 0 || mask.AlphaAt(x-1, y).A == 0 ||
+					mask.AlphaAt(x, y+1).A == 0 || mask.AlphaAt(x, y-1).A == 0 {
+					outline.SetAlpha(x, y, color.Alpha{A: 255})
+				}
+			}
+		}
+	}
+	return outline
+}
+
 func main() {
 	bgPath := "bg-1.png"
 	slicePath := "slice-1.png"
@@ -222,9 +239,9 @@ func main() {
 	mask := ExtractMask(sliceRGBA)
 
 	// Apply Gaussian Blur (to reduce noise)
-	// Radius 2.0, Sigma 1.0
-	bgBlurred := blur.Gaussian(bgRGBA, 2.0)
-	sliceBlurred := blur.Gaussian(sliceRGBA, 2.0)
+	// Radius 3.0, Sigma 1.5
+	bgBlurred := blur.Gaussian(bgRGBA, 3.0)
+	sliceBlurred := blur.Gaussian(sliceRGBA, 3.0)
 
 	// Apply Sobel Edge Detection
 	bgGray := SobelEdgeDetection(bgBlurred)