package gosdk
|
|
/*
|
#cgo CFLAGS: -I. -I./sdk/face/include -I./sdk/darknet/include -I/usr/local/cuda/include -w -g
|
#cgo CXXFLAGS: -I. -I./sdk/face/include -I./sdk/darknet/include -I/usr/local/cuda/include -w -g -std=c++11
|
#cgo LDFLAGS: -L/usr/local/cuda/lib64 -L${SRCDIR}/sdk/face/lib/gpu -L${SRCDIR}/sdk/darknet/lib
|
#cgo LDFLAGS: -Wl,-rpath,${SRCDIR}/sdk/face/lib/gpu:${SRCDIR}/sdk/darknet/lib
|
#cgo LDFLAGS: -ldarknet -lTHFaceImage -lTHFeature -lTHFaceProperty -lTHFaceTracking -lcudart -lcublas -lcurand -lrt -ldl -lpthread
|
#include <stdlib.h>
|
#include "csdk.h"
|
*/
|
import "C"
|
import (
|
"unsafe"
|
)
|
|
// YoloHandle wrap C
|
type YoloHandle struct {
|
handle C.YoloHandle
|
}
|
|
// SDKImage sdk image
|
type SDKImage struct {
|
Data []byte
|
Width int
|
Height int
|
}
|
|
// InitYolo init yolo sdk
|
func InitYolo(fc, fw, fn string, gi int) *YoloHandle {
|
|
c := C.CString(fc)
|
defer C.free(unsafe.Pointer(c))
|
w := C.CString(fw)
|
defer C.free(unsafe.Pointer(w))
|
n := C.CString(fn)
|
defer C.free(unsafe.Pointer(n))
|
|
g := C.int(gi)
|
|
p := C.c_api_yolo_init(c, w, n, g)
|
return &YoloHandle{p}
|
}
|
|
// InitFaceDetector init face detector
|
func InitFaceDetector(tm, gi, minFaces, rollAngle int) {
|
|
C.c_api_face_detector_init(C.int(tm), C.int(gi), C.int(minFaces), C.int(rollAngle))
|
}
|
|
// InitFaceProperty init face property
|
func InitFaceProperty(tm int) {
|
|
C.c_api_face_property_init(C.int(tm))
|
}
|
|
// InitFaceExtractor init face extractor
|
func InitFaceExtractor(tm, gi int) {
|
C.c_api_face_extractor_init(C.int(tm), C.int(gi))
|
}
|
|
// InitFaceTracker init face tracker
|
func InitFaceTracker(tm, gi, w, h, maxFaces, interval, sample int) {
|
|
C.c_api_face_tracker_init(C.int(tm), C.int(gi), C.int(w), C.int(h), C.int(maxFaces), C.int(interval), C.int(sample))
|
}
|
|
// Free free sdk
|
func Free() {
|
C.c_api_release()
|
}
|
|
// FaceDetect face detect
|
func FaceDetect(img SDKImage, ch int) []CFacePos {
|
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
var count C.int
|
cfpos := C.c_api_face_detect(&count, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
|
if cfpos != nil {
|
defer C.free(unsafe.Pointer(cfpos))
|
return CFacePosArrayToGoArray(unsafe.Pointer(cfpos), int(count))
|
}
|
return nil
|
}
|
|
// FaceProperty face property
|
func FaceProperty(fpos CFacePos, img SDKImage, ch int) CThftResult {
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
pos := (*C.cFacePos)(unsafe.Pointer(&fpos))
|
result := C.c_api_face_property(pos, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
|
|
return *((*CThftResult)(unsafe.Pointer(&result)))
|
}
|
|
// FaceExtract face extract feature
|
func FaceExtract(fpos CFacePos, img SDKImage, ch int) []byte {
|
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
var featLen C.int
|
pos := (*C.cFacePos)(unsafe.Pointer(&fpos))
|
|
p := C.c_api_face_extract(&featLen, pos, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
|
defer C.free(unsafe.Pointer(p))
|
b := C.GoBytes(unsafe.Pointer(p), featLen)
|
return b
|
}
|
|
// FaceCompare face compare
|
func FaceCompare(feat1 []byte, feat2 []byte) float32 {
|
res := C.c_api_face_compare((*C.uchar)(unsafe.Pointer(&feat1[0])), (*C.uchar)(unsafe.Pointer(&feat2[0])))
|
return float32(res)
|
}
|
|
// FaceTrackOnly face tracker face only
|
func FaceTrackOnly(img SDKImage, ch int) []CRECT {
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
var fCount C.int
|
rect := C.c_api_face_track_only(&fCount, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
|
if rect != nil {
|
defer C.free(unsafe.Pointer(rect))
|
return CRECTArrayToGoArray(unsafe.Pointer(rect), int(fCount))
|
}
|
return nil
|
}
|
|
// FaceTrackDetect face tracker face detect
|
func FaceTrackDetect(img SDKImage, ch int) []CFaceInfo {
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
var fCount C.int
|
finfo := C.c_api_face_track_only(&fCount, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
|
if finfo != nil {
|
defer C.free(unsafe.Pointer(finfo))
|
return CFaceInfoArrayToGoArray(unsafe.Pointer(finfo), int(fCount))
|
}
|
return nil
|
}
|
|
// FaceTrackingInfo face track info
|
type FaceTrackingInfo struct {
|
tracking bool
|
faces []CFaceInfo
|
}
|
|
// FaceTrack face tracking info
|
func FaceTrack(res map[int64]FaceTrackingInfo, img SDKImage, ch int) {
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
var fCount C.int
|
cFinfo := C.c_api_face_track(&fCount, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.int(ch))
|
if cFinfo == nil {
|
return
|
}
|
defer C.free(unsafe.Pointer(cFinfo))
|
goFinfo := CFaceInfoArrayToGoArray(unsafe.Pointer(cFinfo), int(fCount))
|
|
// 空，添加
|
if len(res) == 0 {
|
for _, v := range goFinfo {
|
var faces []CFaceInfo
|
faces = append(faces, v)
|
tInfo := FaceTrackingInfo{true, faces}
|
res[v.NFaceID] = tInfo
|
}
|
// 非空
|
} else {
|
// 默认所有faces没有tracking
|
for _, v := range res {
|
v.tracking = false
|
}
|
// 查找是否存在tracking
|
for _, v := range goFinfo {
|
// 存在已有信息
|
if info, ok := res[v.NFaceID]; ok {
|
// tracking到face
|
info.tracking = true
|
// 如果tracking长度大于100，删除一些
|
if len(info.faces) > 100 {
|
info.faces = info.faces[1:]
|
}
|
info.faces = append(info.faces, v)
|
// 不存在，新face，加入
|
} else {
|
var faces []CFaceInfo
|
faces = append(faces, v)
|
tInfo := FaceTrackingInfo{true, faces}
|
res[v.NFaceID] = tInfo
|
}
|
}
|
// 删除不再tracking的faces
|
for k, v := range res {
|
if !v.tracking {
|
delete(res, k)
|
}
|
}
|
}
|
}
|
|
// YoloDetect yolo detect
|
func YoloDetect(handle *YoloHandle, img SDKImage, thrsh float32, umns int) []CObjInfo {
|
|
data := img.Data
|
w := img.Width
|
h := img.Height
|
|
var count C.int
|
|
cobjinfo := C.c_api_yolo_detect(handle.handle, &count, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h), C.float(thrsh), C.int(umns))
|
|
if cobjinfo != nil {
|
defer C.free(unsafe.Pointer(cobjinfo))
|
return CYoloObjInfoArrayToGoArray(unsafe.Pointer(cobjinfo), int(count))
|
}
|
return nil
|
}
|
|
// YoloObjName obj name by type
|
func YoloObjName(typ int) string {
|
p := C.c_api_yolo_obj_name(C.int(typ))
|
|
return C.GoString(p)
|
}
|
