recover

zhangmeng

2019-10-21 f3c1f5718e11d31ea342520b61ad7974774b5d1a

 gosdk.go

@@ -84,6 +84,7 @@
   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
@@ -112,6 +113,7 @@
   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
}
@@ -120,6 +122,36 @@
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
@@ -139,6 +171,7 @@
   if cFinfo == nil {
      return
   }
   defer C.free(unsafe.Pointer(cFinfo))
   goFinfo := CFaceInfoArrayToGoArray(unsafe.Pointer(cFinfo), int(fCount))

   // 空，添加
@@ -211,6 +244,7 @@
   if cFinfo == nil {
      return faces
   }
   defer C.free(unsafe.Pointer(cFinfo))
   faces = CFaceInfoArrayToGoArray(unsafe.Pointer(cFinfo), int(fCount))
   //if len(faces) > 0{
   // fmt.Println("faces detected:", len(faces))
@@ -231,6 +265,7 @@
   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