apps/pigo-wasm-demos
1
0
Fork 0
mirror of https://github.com/esimov/pigo-wasm-demos.git synced 2025-09-26 20:31:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fixed issue applying the blur effect on multiple face detection

Browse Source
This commit is contained in:
esimov
2022-05-26 14:23:22 +03:00
parent e93eae9fb9
commit c268418b79
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
bgblur/canvas.go
View File

@@ -149,7 +149,7 @@ func (c *Canvas) Render() error {
gray := pixels.RgbaToGrayscale(data, width, height)
res := pigo.DetectFaces(gray, height, width)
if err := c.drawDetection(res); err != nil {
if err := c.drawDetection(res, imageData); err != nil {
return err
}
}
@@ -248,7 +248,7 @@ func (c *Canvas) blurBackground(src image.Image) (*image.NRGBA, error) {
}
// drawDetection draws the detected faces and eyes.
func (c *Canvas) drawDetection(dets [][]int) error {
func (c *Canvas) drawDetection(dets [][]int, imageData js.Value) error {
var scaleX, scaleY, invScaleX, invScaleY float64
var grad js.Value
@@ -288,6 +288,9 @@ func (c *Canvas) drawDetection(dets [][]int) error {
c.ctxEllipse.Set("fillStyle", grad)
c.ctxEllipse.Call("fillRect", 0, 0, float64(scale)*invScaleX, float64(scale)*invScaleY)
// Replace the underlying face region with the original image.
c.ctxFace.Call("putImageData", imageData, 0, 0)
// Calculate the lean angle between the eyes.
angle := 1 - (math.Atan2(float64(rightPupil.Col-leftPupil.Col), float64(rightPupil.Row-leftPupil.Row)) * 180 / math.Pi / 90)
@@ -297,7 +300,7 @@ func (c *Canvas) drawDetection(dets [][]int) error {
c.ctxFace.Call("translate", float64(-scale)*invScaleX, float64(-scale)*invScaleY)
// Apply the ellipse mask over the source image by using composite operation.
c.ctxFace.Set("globalCompositeOperation", "destination-atop")
c.ctxFace.Set("globalCompositeOperation", "destination-in")
c.ctxFace.Call("drawImage", c.ellipse, row-scale/2, col-scale/2)
c.ctxFace.Call("restore")