-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathplugin.go
35 lines (31 loc) · 1.07 KB
/
plugin.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
package plugin
import (
"gopkg.in/sensorbee/opencv.v0"
"gopkg.in/sensorbee/sensorbee.v0/bql"
"gopkg.in/sensorbee/sensorbee.v0/bql/udf"
)
// initialize scouter components. this init method will be called by
// SensorBee customized main.go.
//
// import(
// _ "gopkg.in/sensorbee/opencv.v0/plugin"
// )
func init() {
// capture
bql.MustRegisterGlobalSourceCreator("opencv_capture_from_uri",
&opencv.FromURICreator{})
bql.MustRegisterGlobalSourceCreator("opencv_capture_from_device",
&opencv.FromDeviceCreator{})
// cascade classifier
udf.MustRegisterGlobalUDSCreator("opencv_cascade_classifier",
udf.UDSCreatorFunc(opencv.NewCascadeClassifier))
udf.MustRegisterGlobalUDF("opencv_detect_multi_scale",
udf.MustConvertGeneric(opencv.DetectMultiScale))
udf.MustRegisterGlobalUDF("opencv_draw_rects",
udf.MustConvertGeneric(opencv.DrawRectsToImage))
// mount image
udf.MustRegisterGlobalUDSCreator("opencv_shared_image",
udf.UDSCreatorFunc(opencv.NewSharedImage))
udf.MustRegisterGlobalUDF("opencv_mount_image",
udf.MustConvertGeneric(opencv.MountAlphaImage))
}