facemark support

.gitignore

@@ -10,3 +10,4 @@ count.out
 .idea/
 contrib/data.yaml
 contrib/testOilPainting.png
+lbfmodel.yaml

contrib/facemarkLBF.cpp

@@ -0,0 +1,23 @@
+#include "facemarkLBF.h"
+
+LBPHFaceMark CreateLBPHFaceMark() {
+    return new cv::Ptr<cv::face::Facemark>(cv::face::FacemarkLBF::create());
+}
+
+void LBPHFaceMark_LoadModel(LBPHFaceMark fm, const char*  model) {
+    (*fm)->loadModel(model);
+}
+
+bool LBPHFaceMark_Fit(LBPHFaceMark fm, Mat frame, struct Rects faces, Points2fVector landmarks) {
+    std::vector<cv::Rect> _faces;
+
+    for (int i = 0; i < faces.length; ++i) {
+        _faces.push_back(cv::Rect(
+            faces.rects[i].x,
+            faces.rects[i].y,
+            faces.rects[i].width,
+            faces.rects[i].height
+        ));
+    }
+    return (*fm)->fit(*frame, _faces, *landmarks);
+}

contrib/facemarkLBF.go

@@ -0,0 +1,57 @@
+package contrib
+
+/*
+#include <stdlib.h>
+#include "facemarkLBF.h"
+#include "../core.h"
+*/
+import "C"
+import (
+	"gocv.io/x/gocv"
+	"image"
+	"unsafe"
+)
+
+type LBPHFaceMark struct {
+	p C.LBPHFaceMark
+}
+
+func NewLBPHFaceMark() *LBPHFaceMark {
+	return &LBPHFaceMark{p: C.CreateLBPHFaceMark()}
+}
+
+func (mark *LBPHFaceMark) LoadModel(model string) {
+	cName := C.CString(model)
+	defer C.free(unsafe.Pointer(cName))
+	C.LBPHFaceMark_LoadModel(mark.p, cName)
+}
+
+// LBPHFaceMark_Fit(LBPHFaceMark fm, Mat frame, struct Rects faces, Points2fVector landmarks)
+func (mark *LBPHFaceMark) Fit(img gocv.Mat, faceBox []image.Rectangle) (bool, [][]gocv.Point2f) {
+
+	bboxesRectArr := []C.struct_Rect{}
+	for _, v := range faceBox {
+		bbox := C.struct_Rect{
+			x:      C.int(v.Min.X),
+			y:      C.int(v.Min.Y),
+			width:  C.int(v.Size().X),
+			height: C.int(v.Size().Y),
+		}
+		bboxesRectArr = append(bboxesRectArr, bbox)
+	}
+
+	bboxesRects := C.Rects{
+		rects:  (*C.Rect)(&bboxesRectArr[0]),
+		length: C.int(len(faceBox)),
+	}
+
+	points2fVector := gocv.NewPoints2fVector()
+	result := bool(C.LBPHFaceMark_Fit(
+		mark.p,
+		C.Mat(img.Ptr()),
+		bboxesRects,
+		C.Points2fVector(points2fVector.P()),
+	))
+
+	return result, points2fVector.ToPoints()
+}

contrib/facemarkLBF.h

@@ -0,0 +1,32 @@
+#ifndef _OPENCV3_FACE_H_
+#define _OPENCV3_FACE_H_
+
+#ifdef __cplusplus
+#include <opencv2/opencv.hpp>
+#include <opencv2/face.hpp>
+
+extern "C" {
+#endif
+
+#include "../core.h"
+
+#ifdef __cplusplus
+typedef cv::Ptr<cv::face::Facemark>* LBPHFaceMark;
+#else
+typedef void* LBPHFaceMark;
+#endif
+
+struct PredictResponse {
+    int label;
+    double confidence;
+};
+
+LBPHFaceMark CreateLBPHFaceMark();
+void LBPHFaceMark_LoadModel(LBPHFaceMark fm, const char*  model); //Points2fVector
+bool LBPHFaceMark_Fit(LBPHFaceMark fm, Mat frame, struct Rects faces, Points2fVector landmarks);
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif //_OPENCV3_FACE_H_

contrib/facemarkLBF_test.go

@@ -0,0 +1,21 @@
+package contrib
+
+import (
+	"log"
+	"testing"
+)
+
+func TestNewLBPHFaceMark(t *testing.T) {
+	mark := NewLBPHFaceMark()
+	log.Println(mark)
+}
+
+func TestLBPHFaceMark_LoadModel(t *testing.T) {
+	mark := NewLBPHFaceMark()
+	log.Println(mark)
+	mark.LoadModel("/Users/wushaojie/Documents/project/golang/go-opencv/lbfmodel.yaml")
+	log.Println("success")
+}
+
+func TestLBPHFaceMark_Fit(t *testing.T) {
+}