forked from go-opencv/go-opencv
-
Notifications
You must be signed in to change notification settings - Fork 0
/
webcam.go
75 lines (59 loc) · 1.48 KB
/
webcam.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package main
import (
"fmt"
"os"
//"path"
//"runtime"
"github.com/lazywei/go-opencv/opencv"
//"../opencv" // can be used in forks, comment in real application
)
func main() {
var edge_threshold int
win := opencv.NewWindow("Go-OpenCV Webcam")
defer win.Destroy()
cap := opencv.NewCameraCapture(0)
if cap == nil {
panic("can not open camera")
}
defer cap.Release()
win.CreateTrackbar("Thresh", 1, 100, func(pos int, param ...interface{}) {
edge_threshold = pos
})
fmt.Println("Press ESC to quit")
for {
if cap.GrabFrame() {
img := cap.RetrieveFrame(1)
if img != nil {
ProcessImage(img, win, edge_threshold)
} else {
fmt.Println("Image ins nil")
}
}
key := opencv.WaitKey(10)
if key == 27 {
os.Exit(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(cedge)
return nil
}