hybridgroup.gobot/examples/opencv_face_detect.go

56 lines
973 B
Go

// +build example
//
// Do not build by default.
package main
import (
"path"
"runtime"
"sync/atomic"
"time"
"github.com/hybridgroup/gocv"
"gobot.io/x/gobot"
"gobot.io/x/gobot/platforms/opencv"
)
var img atomic.Value
func main() {
_, currentfile, _, _ := runtime.Caller(0)
cascade := path.Join(path.Dir(currentfile), "haarcascade_frontalface_alt.xml")
window := opencv.NewWindowDriver()
camera := opencv.NewCameraDriver(1)
work := func() {
mat := gocv.NewMat()
img.Store(mat)
camera.On(opencv.Frame, func(data interface{}) {
i := data.(gocv.Mat)
img.Store(i)
})
gobot.Every(10*time.Millisecond, func() {
i := img.Load().(gocv.Mat)
if i.Empty() {
return
}
faces := opencv.DetectFaces(cascade, i)
opencv.DrawRectangles(i, faces, 0, 255, 0, 5)
window.ShowImage(i)
window.WaitKey(1)
})
}
robot := gobot.NewRobot("faceBot",
[]gobot.Connection{},
[]gobot.Device{window, camera},
work,
)
robot.Start()
}