apps/pigo-wasm-demos
1
0
Fork 0
mirror of https://github.com/esimov/pigo-wasm-demos.git synced 2025-12-24 12:47:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fine tuning and adding new features

Browse Source
This commit is contained in:
esimov
2020-04-29 07:47:58 +03:00
parent 09b42099b8
commit e770770a6c
5 changed files with 56 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
index.html
View File

@@ -26,7 +26,7 @@
<body>
<script type="text/javascript">
var stats = new Stats();
stats.showPanel(0);
stats.showPanel(1);
document.body.appendChild(stats.dom);
</script>
</body>

BIN
lib.wasm
View File

Binary file not shown.

80
pixelate/canvas.go
View File

@@ -33,8 +33,9 @@ type Canvas struct {
video js.Value
// Canvas interaction related variables
showPupil bool
drawCircle bool
showPupil bool
showFrame bool
useNoise bool
// Quantizer related variables
numOfColors int
@@ -48,6 +49,11 @@ type SubImager interface {
SubImage(r image.Rectangle) image.Image
}
const (
maxColors = 32
minColors = 2
)
var (
det *detector.Detector
quant Quant
@@ -71,9 +77,10 @@ func NewCanvas() *Canvas {
c.ctx = c.canvas.Call("getContext", "2d")
c.showPupil = false
c.drawCircle = false
c.showFrame = false
c.useNoise = false
c.numOfColors = 32
c.numOfColors = 8
c.cellSize = 10
det = detector.NewDetector()
@@ -239,13 +246,13 @@ func (c *Canvas) imgToPix(img image.Image) []uint8 {
}
// pixelateDetectedRegion pixelates the detected face region
func (c *Canvas) pixelateDetectedRegion(data []uint8, dets []int) []uint8 {
func (c *Canvas) pixelateDetectedRegion(data []uint8, dets []int, useNoise bool) []uint8 {
// Converts the array buffer to an image
img := c.pixToImage(data, dets[2])
img := c.pixToImage(data, int(float64(dets[2])*0.75))
// Quantize the substracted image in order to reduce the number of colors.
// This will results in a pixelated subtype image.
cell := quant.Draw(img, c.numOfColors, c.cellSize)
cell := quant.Draw(img, c.numOfColors, c.cellSize, useNoise)
return c.imgToPix(cell)
}
@@ -254,32 +261,31 @@ func (c *Canvas) drawDetection(data []uint8, dets [][]int) {
for i := 0; i < len(dets); i++ {
if dets[i][3] > 50 {
c.ctx.Call("beginPath")
c.ctx.Set("lineWidth", 3)
c.ctx.Set("strokeStyle", "red")
c.ctx.Set("lineWidth", 2)
c.ctx.Set("strokeStyle", "rgba(255, 0, 0, 0.5)")
row, col, scale := dets[i][1], dets[i][0], dets[i][2]
if c.drawCircle {
c.ctx.Call("moveTo", row+int(scale/2), col)
c.ctx.Call("arc", row, col, scale/2, 0, 2*math.Pi, true)
} else {
col = col + int(float64(col)*0.125)
scale = int(float64(scale) * 0.75)
if c.showFrame {
c.ctx.Call("rect", row-scale/2, col-scale/2, scale, scale)
// Substract the image under the detected face region.
imgData := make([]byte, scale*scale*4)
subimg := c.ctx.Call("getImageData", row-scale/2, col-scale/2, scale, scale).Get("data")
uint8Arr := js.Global().Get("Uint8Array").New(subimg)
js.CopyBytesToGo(imgData, uint8Arr)
buffer := c.pixelateDetectedRegion(imgData, dets[i])
uint8Arr = js.Global().Get("Uint8Array").New(scale * scale * 4)
js.CopyBytesToJS(uint8Arr, buffer)
uint8Clamped := js.Global().Get("Uint8ClampedArray").New(uint8Arr)
rawData := js.Global().Get("ImageData").New(uint8Clamped, scale)
// Replace the underlying face region byte array with the quantized values.
c.ctx.Call("putImageData", rawData, row-scale/2, col-scale/2)
}
// Substract the image under the detected face region.
imgData := make([]byte, scale*scale*4)
subimg := c.ctx.Call("getImageData", row-scale/2, col-scale/2, scale, scale).Get("data")
uint8Arr := js.Global().Get("Uint8Array").New(subimg)
js.CopyBytesToGo(imgData, uint8Arr)
buffer := c.pixelateDetectedRegion(imgData, dets[i], c.useNoise)
uint8Arr = js.Global().Get("Uint8Array").New(scale * scale * 4)
js.CopyBytesToJS(uint8Arr, buffer)
uint8Clamped := js.Global().Get("Uint8ClampedArray").New(uint8Arr)
rawData := js.Global().Get("ImageData").New(uint8Clamped, scale)
// Replace the underlying face region byte array with the quantized values.
c.ctx.Call("putImageData", rawData, row-scale/2, col-scale/2)
c.ctx.Call("stroke")
if c.showPupil {
@@ -309,10 +315,20 @@ func (c *Canvas) detectKeyPress() {
switch {
case keyCode.String() == "s":
c.showPupil = !c.showPupil
case keyCode.String() == "c":
c.drawCircle = !c.drawCircle
case keyCode.String() == "f":
c.showFrame = !c.showFrame
case keyCode.String() == "n":
c.useNoise = !c.useNoise
case keyCode.String() == "=":
if c.numOfColors <= maxColors {
c.numOfColors++
}
case keyCode.String() == "-":
if c.numOfColors > minColors {
c.numOfColors--
}
default:
c.drawCircle = false
c.showFrame = false
}
return nil
})

11
pixelate/drawer.go
View File

@@ -13,10 +13,10 @@ type context struct {
}
// Brightness factor
var bf = 1.0
var bf = 1.0005
// Draw creates uniform cells with the quantified cell color of the source image.
func (quant *Quant) Draw(img image.Image, numOfColors int, csize int) image.Image {
func (quant *Quant) Draw(img image.Image, numOfColors int, csize int, useNoise bool) image.Image {
var cellSize int
dx, dy := img.Bounds().Dx(), img.Bounds().Dy()
@@ -56,9 +56,10 @@ func (quant *Quant) Draw(img image.Image, numOfColors int, csize int) image.Imag
}
}
ctxImg := ctx.Image()
noisyImg := noise(10, ctxImg, dx, dy)
return noisyImg
if useNoise {
return noise(ctxImg, dx, dy, 12)
}
return ctxImg
}
// drawCell draws the cell filling up with the quantified color

2
pixelate/seed.go
View File

@@ -14,7 +14,7 @@ type prng struct {
}
// noise apply a noise factor to the source image
func noise(amount int, pxl image.Image, w, h int) *image.NRGBA64 {
func noise(pxl image.Image, w, h int, amount int) *image.NRGBA64 {
noiseImg := image.NewNRGBA64(image.Rect(0, 0, w, h))
prng := &prng{
a: 16807,