contrib/face: extra setters and getters for LBPHFaceRecognizer (#1194)

contrib/face: extra setters and getters for LBPHFaceRecognizer

Author: diegohce

contrib/face.cpp

@@ -60,6 +60,10 @@ struct PredictResponse LBPHFaceRecognizer_PredictExtended(LBPHFaceRecognizer fr,
     return response;
 }
 
+double LBPHFaceRecognizer_GetThreshold(LBPHFaceRecognizer fr){
+    return (*fr)->getThreshold();
+}
+
 void LBPHFaceRecognizer_SetThreshold(LBPHFaceRecognizer fr, double threshold) {
     (*fr)->setThreshold(threshold);
 
@@ -99,4 +103,26 @@ void LBPHFaceRecognizer_LoadFile(LBPHFaceRecognizer fr, const char*  filename) {
     return;
 }
 
+void LBPHFaceRecognizer_SetGridX(LBPHFaceRecognizer fr, int x) {
+    (*fr)->setGridX(x);
+
+    return;
+}
+
+void LBPHFaceRecognizer_SetGridY(LBPHFaceRecognizer fr, int y) {
+    (*fr)->setGridY(y);
+
+    return;
+}
+
+int LBPHFaceRecognizer_GetGridX(LBPHFaceRecognizer fr) {
+    int n = (*fr)->getGridX();
+
+    return n;
+}
+
+int LBPHFaceRecognizer_GetGridY(LBPHFaceRecognizer fr) {
+    int n = (*fr)->getGridY();
+    return n;
+}
 

contrib/face.go

@@ -6,6 +6,7 @@ package contrib
 */
 import "C"
 import (
+	"image"
 	"unsafe"
 
 	"gocv.io/x/gocv"
@@ -107,6 +108,16 @@ func (fr *LBPHFaceRecognizer) PredictExtendedResponse(sample gocv.Mat) PredictRe
 	return resp
 }
 
+// GetThreshold gets the threshold value of the model, i.e. the threshold
+// applied in the prediction.
+//
+// For further information, see:
+// https://docs.opencv.org/4.x/df/d25/classcv_1_1face_1_1LBPHFaceRecognizer.html#acf2a6993eb4347b3f89009da693a3f70
+func (fr *LBPHFaceRecognizer) GetThreshold() float32 {
+	t := C.LBPHFaceRecognizer_GetThreshold(fr.p)
+	return float32(t)
+}
+
 // SetThreshold sets the threshold value of the model, i.e. the threshold
 // applied in the prediction.
 //
@@ -164,3 +175,48 @@ func (fr *LBPHFaceRecognizer) LoadFile(fname string) {
 	defer C.free(unsafe.Pointer(cName))
 	C.LBPHFaceRecognizer_LoadFile(fr.p, cName)
 }
+
+// SetGridX sets grid's X value
+//
+// For further information, see:
+// https://docs.opencv.org/4.x/df/d25/classcv_1_1face_1_1LBPHFaceRecognizer.html#ad65975baee31dbf3bd2a750feef74831
+func (fr *LBPHFaceRecognizer) SetGridX(x int) {
+	C.LBPHFaceRecognizer_SetGridX(fr.p, C.int(x))
+}
+
+// SetGridY sets grid's Y value
+//
+// For further information, see:
+// https://docs.opencv.org/4.x/df/d25/classcv_1_1face_1_1LBPHFaceRecognizer.html#a9cebb0138dbb3553b27beb2df3924ae6
+func (fr *LBPHFaceRecognizer) SetGridY(y int) {
+	C.LBPHFaceRecognizer_SetGridY(fr.p, C.int(y))
+}
+
+// GetGridX gets grid's X value
+//
+// For further information, see:
+// https://docs.opencv.org/4.x/df/d25/classcv_1_1face_1_1LBPHFaceRecognizer.html#ada6839bed931a8f68c5127e1af7a8b83
+func (fr *LBPHFaceRecognizer) GetGridX() int {
+	x := C.LBPHFaceRecognizer_GetGridX(fr.p)
+	return int(x)
+}
+
+// GetGridY gets grid's Y value
+//
+// For further information, see:
+// https://docs.opencv.org/4.x/df/d25/classcv_1_1face_1_1LBPHFaceRecognizer.html#a22c68c0baf3eb9e852f47ae9241dbf15
+func (fr *LBPHFaceRecognizer) GetGridY() int {
+	y := C.LBPHFaceRecognizer_GetGridY(fr.p)
+	return int(y)
+}
+
+// SetGrid helper for SetGrid* functions
+func (fr *LBPHFaceRecognizer) SetGrid(p image.Point) {
+	fr.SetGridX(p.X)
+	fr.SetGridY(p.Y)
+}
+
+// GetGrid helper for GetGrid* functions
+func (fr *LBPHFaceRecognizer) GetGrid() image.Point {
+	return image.Point{X: fr.GetGridX(), Y: fr.GetGridY()}
+}

contrib/face.h

@@ -26,12 +26,19 @@ void LBPHFaceRecognizer_Train(LBPHFaceRecognizer fr, Mats images, IntVector labe
 void LBPHFaceRecognizer_Update(LBPHFaceRecognizer fr, Mats images, IntVector labels);
 int LBPHFaceRecognizer_Predict(LBPHFaceRecognizer fr, Mat sample);
 struct PredictResponse LBPHFaceRecognizer_PredictExtended(LBPHFaceRecognizer fr, Mat sample);
+double LBPHFaceRecognizer_GetThreshold(LBPHFaceRecognizer fr);
 void LBPHFaceRecognizer_SetThreshold(LBPHFaceRecognizer fr, double threshold);
 void LBPHFaceRecognizer_SetRadius(LBPHFaceRecognizer fr, int radius);
 void LBPHFaceRecognizer_SetNeighbors(LBPHFaceRecognizer fr, int neighbors);
 void LBPHFaceRecognizer_SaveFile(LBPHFaceRecognizer fr, const char*  filename);
 void LBPHFaceRecognizer_LoadFile(LBPHFaceRecognizer fr, const char*  filename);
 int LBPHFaceRecognizer_GetNeighbors(LBPHFaceRecognizer fr);
+void LBPHFaceRecognizer_SetGridX(LBPHFaceRecognizer fr, int x);
+void LBPHFaceRecognizer_SetGridY(LBPHFaceRecognizer fr, int y);
+int LBPHFaceRecognizer_GetGridX(LBPHFaceRecognizer fr);
+int LBPHFaceRecognizer_GetGridY(LBPHFaceRecognizer fr);
+
+
 
 #ifdef __cplusplus
 }

contrib/face_test.go

@@ -1,9 +1,11 @@
 package contrib
 
 import (
-	"gocv.io/x/gocv"
+	"image"
 	"math"
 	"testing"
+
+	"gocv.io/x/gocv"
 )
 
 func TestLBPHFaceRecognizer_Methods(t *testing.T) {
@@ -90,3 +92,21 @@ func TestLBPHFaceRecognizer_Methods(t *testing.T) {
 		t.Errorf("Invalid loaded data: %d", label)
 	}
 }
+
+func TestLBPHFaceRecognizer_SetGridY_GetGridY(t *testing.T) {
+	facer := NewLBPHFaceRecognizer()
+	facer.SetGridY(5)
+
+	if v := facer.GetGridY(); v != 5 {
+		t.Errorf("got %d want 5", v)
+	}
+}
+
+func TestLBPHFaceRecognizer_SetGrid_GetGrid(t *testing.T) {
+	facer := NewLBPHFaceRecognizer()
+	facer.SetGrid(image.Point{7, 5})
+
+	if p := facer.GetGrid(); p.X != 7 || p.Y != 5 {
+		t.Errorf("got %+v want {7, 5}", p)
+	}
+}