update

zhangmeng

2019-12-16 4217e31b519aa4f3f7e3637e9b43a04052a21092

update

 goface.go

@@ -24,10 +24,11 @@
    extractor   bool
    propertizer bool
    tracker     bool
    fnLogger    func(...interface{})
}

// NewSDK sdk
func NewSDK() interface{} {
func NewSDK(fn func(...interface{})) interface{} {
    h := C.create_sdkface()
    if h == nil {
        return nil
@@ -39,6 +40,7 @@
        extractor:   false,
        propertizer: false,
        tracker:     false,
        fnLogger:    fn,
    }
}

@@ -50,6 +52,12 @@
    }
}

func (s *SDKFace) printLog(l ...interface{}) {
    if s.fnLogger != nil {
        s.fnLogger(l)
    }
}

// Detector detector
func Detector(i interface{}, minFaces, rollAngles, threadMax, gpu int) bool {
    s := i.(*SDKFace)
@@ -58,6 +66,7 @@
    }
    ret := C.init_detector(s.handle, C.int(minFaces), C.int(rollAngles), C.int(threadMax), C.int(gpu))
    if ret <= 0 {
        s.printLog("->face--> CREATE Detector ERROR")
        return false
    }
    s.detector = true
@@ -72,6 +81,7 @@
    }
    ret := C.init_extractor(s.handle, C.int(threadMax), C.int(gpu))
    if ret <= 0 {
        s.printLog("->face--> CREATE Extractor ERROR")
        return false
    }
    s.extractor = true
@@ -86,6 +96,7 @@
    }
    ret := C.init_propertizer(s.handle, C.int(threadMax))
    if ret <= 0 {
        s.printLog("->face--> CREATE Propertizer ERROR")
        return false
    }
    s.propertizer = true
@@ -101,6 +112,7 @@
    ret := C.init_tracker(s.handle, C.int(w), C.int(h), C.int(maxFaces), C.int(interval), C.int(sampleSize), C.int(threadMax), C.int(gpu))

    if ret <= 0 {
        s.printLog("->face--> CREATE Tracker ERROR")
        return false
    }
    s.tracker = true
@@ -133,6 +145,7 @@
    if ret > 0 {
        return CFacePosArrayToGoArray(cfpos, int(count))
    }
    s.printLog("->face--> Detect No One, Ret: ", ret)
    return nil
}

@@ -144,10 +157,11 @@
    //(void *handle, const cFacePos *pos, const void*data, const int w, const int h, const int c, const int chan, void **feat, int *featLen);
    var feat unsafe.Pointer
    var featLen C.int
    p := C.extract(s.handle, pos, unsafe.Pointer(&data[0]), C.int(w), C.int(h), C.int(c), C.int(ch), &feat, &featLen)
    if p > 0 {
    ret := C.extract(s.handle, pos, unsafe.Pointer(&data[0]), C.int(w), C.int(h), C.int(c), C.int(ch), &feat, &featLen)
    if ret > 0 {
        return C.GoBytes(feat, featLen)
    }
    s.printLog("->face--> Extract Nothing, Ret: ", ret)
    return nil
}

@@ -177,6 +191,7 @@
        C.free(thft)
        return &gothft
    }
    s.printLog("->face--> Propertize Nothing, Ret: ", ret)
    return nil
}

@@ -212,6 +227,7 @@

        return faces
    }
    s.printLog("->face--> Track No One, Ret: ", ret)
    return nil
}

@@ -237,12 +253,14 @@
    s := i.(*SDKFace)

    if !s.tracker {
        s.printLog("->face--> TrackerResize Failed, No Tracker Init")
        return false
    }
    ret := C.resize(s.handle, C.int(w), C.int(h), C.int(ch))
    if ret >= 0 {
    if ret == 0 {
        return true
    }
    s.printLog("->face--> TrackerResize Failed, Ret: ", ret, " SDK Channel: ", ch, " Size: ", w, "x", h)
    return false
}

@@ -263,11 +281,13 @@
        channel = 3
    }

    if !TrackerResize(i, w, h, dchan) {
        return nil
    }

    var fInfo []sdkstruct.CFaceInfo

    if s.tracker {
        fInfo = Track(s, data, w, h, c, dchan)
    }
    fInfo = Track(s, data, w, h, c, dchan)

    var faces []sdkstruct.CFaceResult
    //将sdk返回值转换成protomsg类型
@@ -286,6 +306,7 @@
        faces = append(faces, result)

    }
    s.printLog("->face--> Run Detect Face Count: ", len(fInfo), " Result: ", len(faces))

    return faces
}