hybridgroup.gobot/examples/opencv_face_detect.go

package main

import (
	cv "github.com/hybridgroup/go-opencv/opencv"
	"github.com/hybridgroup/gobot"
	"github.com/hybridgroup/gobot/opencv"
	"path"
	"runtime"
)

func main() {
	_, currentfile, _, _ := runtime.Caller(0)
	cascade := path.Join(path.Dir(currentfile), "haarcascade_frontalface_alt.xml")

	window := opencv.NewWindowDriver()
	window.Name = "window"

	camera := opencv.NewCameraDriver()
	camera.Name = "camera"

	work := func() {
		var image *cv.IplImage
		gobot.On(camera.Events["Frame"], func(data interface{}) {
			image = data.(*cv.IplImage)
		})

		go func() {
			for {
				if image != nil {
					i := image.Clone()
					faces := opencv.DetectFaces(cascade, i)
					i = opencv.DrawRectangles(i, faces, 0, 255, 0, 5)
					window.ShowImage(i)
				}
			}
		}()
	}

	robot := gobot.Robot{
		Connections: []gobot.Connection{},
		Devices:     []gobot.Device{window, camera},
		Work:        work,
	}

	robot.Start()
}