update

zhangmeng

2019-11-15 4d025c4a8c81c0bfdd159586528db0af72e6a6d6

 gosdk.go

@@ -1,11 +1,11 @@
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 CFLAGS: -I. -I./sdk/face/include -I./sdk/darknet/include -I/usr/local/cuda/include -I./sdk/plate/include -w -g
#cgo CXXFLAGS: -I. -I./sdk/face/include -I./sdk/darknet/include -I/usr/local/cuda/include -I./sdk/plate/include -w -g -std=c++11
#cgo LDFLAGS: -L/usr/local/cuda/lib64 -L${SRCDIR}/sdk/face/lib/gpu -L${SRCDIR}/sdk/darknet/lib -L${SRCDIR}/sdk/plate/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
#cgo LDFLAGS: -ldarknet -lTHFaceImage -lTHFeature -lTHFaceProperty -lTHFaceTracking -lcudart -lcublas -lcurand -lrt -ldl -lpthread -lthplateid
#include <stdlib.h>
#include "csdk.h"
*/
@@ -16,8 +16,12 @@

// YoloHandle wrap C
type YoloHandle struct {
   handle C.YoloHandle
   handle       C.YoloHandle
   LastYoloObjs []CObjTrackInfo //yolo跟踪的上一帧信息
   LastTrackID  uint64          //yolo 被使用的ID
}

const RatioInterTrack = 50 //跟踪判断重叠阈值

// SDKImage sdk image
type SDKImage struct {
@@ -25,10 +29,6 @@
   Width  int
   Height int
}

var LastYoloObjs []CObjTrackInfo //yolo跟踪的上一帧信息
var LastTrackID uint64 = 0       //yolo 被使用的ID
const RatioInterTrack = 50       //跟踪判断重叠阈值

// InitYolo init yolo sdk
func InitYolo(fc, fw, fn string, gi int) *YoloHandle {
@@ -43,7 +43,7 @@
   g := C.int(gi)

   p := C.c_api_yolo_init(c, w, n, g)
   return &YoloHandle{p}
   return &YoloHandle{handle: p}
}

// InitFaceDetector init face detector
@@ -67,6 +67,12 @@
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))
}

// ResizeFaceTracker init face tracker
func ResizeFaceTracker(ch, w, h int) int {

   return int(C.c_api_face_track_resize(C.int(ch), C.int(w), C.int(h)))
}

// Free free sdk
@@ -216,6 +222,43 @@
   }
}

func FaceInfo2FacePos(face CFaceInfo) (fPos CFacePos) {
   fPos.RcFace = face.RcFace
   fPos.PtLeftEye = face.PtLeftEye
   fPos.PtRightEye = face.PtRightEye
   fPos.PtNose = face.PtNose
   fPos.PtMouth = face.PtMouth
   fPos.FAngle.Yaw = face.FAngle.Yaw
   fPos.FAngle.Pitch = face.FAngle.Pitch
   fPos.FAngle.Roll = face.FAngle.Roll
   fPos.FAngle.Confidence = face.FAngle.Confidence

   copy(fPos.PFacialData[:], face.PFacialData[:512])

   return fPos
}

// FaceTrackSimple face tracking info
func FaceTrackSimple(img SDKImage, ch int) (faces []CFaceInfo) {
   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))
   // fmt.Println("cFinfo detected:", cFinfo)
   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))
   //}

   return faces
}

// YoloDetect yolo detect
func YoloDetect(handle *YoloHandle, img SDKImage, thrsh float32, umns int) []CObjInfo {

@@ -288,14 +331,8 @@
   var tmp CObjTrackInfo
   //LastYoloObjs
   detectObjs := YoloDetect(handle, img, thrsh, umns)
   for i := 0; i < len(detectObjs); i++ {
      if detectObjs[i].Typ != 0 {
         detectObjs = append(detectObjs[:i], detectObjs[i+1:]...) //从检测目标里删除已经查到的跟踪目标
         i--
      }
   }

   for _, vLast := range LastYoloObjs {
   for _, vLast := range handle.LastYoloObjs {
      for i := 0; i < len(detectObjs); i++ {
         //fmt.Println("vNew.Typ:", vNew.Typ)
         if vLast.ObjInfo.Typ == detectObjs[i].Typ { //同一类别，比如都是人体
@@ -318,8 +355,8 @@
   if len(detectObjs) > 0 {
      for _, vAdd := range detectObjs {
         tmp.ObjInfo = vAdd
         tmp.ID = LastTrackID
         LastTrackID++
         tmp.ID = handle.LastTrackID
         handle.LastTrackID++

         allObjs = append(allObjs, tmp)
         newObjs = append(newObjs, tmp)
@@ -327,7 +364,110 @@
   }

   //刷新上一帧的跟踪目标
   LastYoloObjs = allObjs
   handle.LastYoloObjs = allObjs

   return allObjs, newObjs
}

// YoloDetectTrack2 yolo detect   (只识别人)
func YoloDetectTrack2(handle *YoloHandle, LastYoloObjs []CObjTrackInfo, LastTrackID *uint64, img SDKImage, thrsh float32, umns int) (allObjs []CObjTrackInfo, newObjs []CObjTrackInfo) {

   var tmp CObjTrackInfo
   //LastYoloObjs
   detectObjs := YoloDetect(handle, img, thrsh, umns)

   for _, vLast := range LastYoloObjs {
      for i := 0; i < len(detectObjs); i++ {
         //fmt.Println("vNew.Typ:", vNew.Typ)
         if vLast.ObjInfo.Typ == detectObjs[i].Typ { //同一类别，比如都是人体
            ratio := countInterAreaOfTwoRect(vLast.ObjInfo.RcObj, detectObjs[i].RcObj)
            if ratio >= RatioInterTrack {
               //update LastYoloObjs
               vLast.ObjInfo.RcObj = detectObjs[i].RcObj
               vLast.ObjInfo.Prob = detectObjs[i].Prob

               allObjs = append(allObjs, vLast)
               detectObjs = append(detectObjs[:i], detectObjs[i+1:]...) //从检测目标里删除已经查到的跟踪目标
               i--
               break //上一帧跟踪的目标已经找到，无需往下处理其他检测目标
            }
         }
      }
   }

   //处理新出现的目标
   id := *LastTrackID
   if len(detectObjs) > 0 {
      for _, vAdd := range detectObjs {
         tmp.ObjInfo = vAdd
         tmp.ID = id
         id++

         allObjs = append(allObjs, tmp)
         newObjs = append(newObjs, tmp)
      }
   }
   *LastTrackID = id
   return allObjs, newObjs
}

func DefaultPlateIDSDKConfig() *CPlateIDCfg{
   return &CPlateIDCfg{
      FastMemorySize: 32,
      MemorySize: 400,
      MinPlateWidth: 60,
      MaxPlateWidth: 400,
      MaxImageWidth: 4096,
      MaxImageHeight: 2160,
      IsFieldImage: 0,
      MovingImage: 1,
      OrderOpt: 0,
      LeanCorrection: 1,
      ImproveSpeed: 0,
      CarLogo: 1,
      LotDetect: 1,
      Shadow: 1,
      ShieldRailing: 1,
      CarModel: 1,
      LocateTh: 5,
      OCRTh: 2,
	
      Individual: 1,
      TwoRowYellow: 1,
      ArmPolice: 1,
      ArmPolice2: 1,
      TwoRowArmy: 1,
      Tractor: 1,
      Embassy: 1,
      ChangNei: 1,
      MinHang: 1,
      Consulate: 1,
      NewEnergy: 1,
      OnlyTwoRowYellow: 0,
      OnlyLocation: 0,
   }
}

// InitPlateIDDetector init plateid detector
func InitPlateIDDetector(config *CPlateIDCfg, soPath []byte) {
   C.c_api_plate_id_init((*C.cPlateIDCfg)(unsafe.Pointer(config)), (*C.char)(unsafe.Pointer(&soPath[0])))
}

// PlateIDDetect plateid detect
func PlateIDDetect(img SDKImage) []CPlateIDResult {
   data := img.Data
   w := img.Width
   h := img.Height

   var count C.int
   cppos := C.c_api_plate_id_detect(&count, (*C.uchar)(unsafe.Pointer(&data[0])), C.int(w), C.int(h))
   if cppos != nil {
      defer C.free(unsafe.Pointer(cppos))
      return CPlateIDPosArrayToGoArray(unsafe.Pointer(cppos), int(count))
   }
   return nil
}

func FreePlateIdDetector() int{
   return int(C.c_api_plate_id_free())
}