Ejemplo n.º 1
0
func ProcessImage(img *opencv.IplImage, win *opencv.Window, pos int) error {
	w := img.Width()
	h := img.Height()

	// Create the output image
	cedge := opencv.CreateImage(w, h, opencv.IPL_DEPTH_8U, 3)
	defer cedge.Release()

	// Convert to grayscale
	gray := opencv.CreateImage(w, h, opencv.IPL_DEPTH_8U, 1)
	edge := opencv.CreateImage(w, h, opencv.IPL_DEPTH_8U, 1)
	defer gray.Release()
	defer edge.Release()

	opencv.CvtColor(img, gray, opencv.CV_BGR2GRAY)

	opencv.Smooth(gray, edge, opencv.CV_BLUR, 3, 3, 0, 0)
	opencv.Not(gray, edge)

	// Run the edge detector on grayscale
	opencv.Canny(gray, edge, float64(pos), float64(pos*3), 3)

	opencv.Zero(cedge)
	// copy edge points
	opencv.Copy(img, cedge, edge)

	win.ShowImage(img)
	return nil
}
Ejemplo n.º 2
0
func ProcessbetterImage(img *opencv.IplImage, win *opencv.Window, pos int) error {
	_, currentfile, _, _ := runtime.Caller(0)
	//image := opencv.LoadImage(path.Join(path.Dir(currentfile), "../images/eye.jpg"))

	if counter == 1 {
		//find eye, cache them and then mark them
		counter = 0
		cachedPointsCenterY = []int{}
		cachedPointsCenterX = []int{}

		if haarClassifier == nil {
			//load and cache haarClassifier
			haarClassifier = opencv.LoadHaarClassifierCascade(path.Join(path.Dir(currentfile), "/cascades/haarcascade_eye_tree_eyeglasses.xml"))
		}

		rects := haarClassifier.DetectObjects(img)

		for _, value := range rects {

			var pointRect1 = opencv.Point{value.X() + value.Width(), value.Y()}
			var pointRect2 = opencv.Point{value.X(), value.Y() + value.Height()}
			var centerPointX = int((pointRect1.X + pointRect2.X) / 2)
			var centerPointY = int((pointRect1.Y + pointRect2.Y) / 2)

			var center = opencv.Point{centerPointX, centerPointY}
			cachedPointsCenterX = append(cachedPointsCenterX, center.X)
			cachedPointsCenterY = append(cachedPointsCenterY, center.Y)

			//draw eye marker
			drawEyeMarker(img, center)
		}
	} else {
		//if we have a cached value
		for i := 0; i < len(cachedPointsCenterY); i++ {
			drawEyeMarker(img, opencv.Point{cachedPointsCenterX[i], cachedPointsCenterY[i]})
		}

	}
	win.ShowImage(img)
	//counter for cache
	counter = counter + 1
	return nil
}