SIFT custom

features2d.cpp

@@ -590,6 +590,11 @@ SIFT SIFT_Create() {
     return new cv::Ptr<cv::SIFT>(cv::SIFT::create());
 }
 
+SIFT SIFT_CreateCustom(int nfeatures, int nOctaveLayers, double contrastThreshold, double edgeThreshold, double sigma) {
+    return new cv::Ptr<cv::SIFT>(cv::SIFT::create(nfeatures, nOctaveLayers, contrastThreshold, edgeThreshold, sigma));
+}
+
+
 void SIFT_Close(SIFT d) {
     delete d;
 }

features2d.go

@@ -943,6 +943,39 @@ func NewSIFT() SIFT {
 	return SIFT{p: unsafe.Pointer(C.SIFT_Create())}
 }
 
+func NewSIFTCustom(nfeatures *int, nOctaveLayers *int, contrastThreshold *float64, edgeThreshold *float64, sigma *float64) SIFT {
+	numFeatures := 0
+	if nfeatures != nil {
+		numFeatures = *nfeatures
+	}
+
+	numOctaveLayers := 3
+
+	if nOctaveLayers != nil {
+		numOctaveLayers = *nOctaveLayers
+	}
+
+	var numContrastThreshold float64 = 0.04
+
+	if contrastThreshold != nil {
+		numContrastThreshold = *contrastThreshold
+	}
+
+	var numEdgeThreshold float64 = 10
+
+	if edgeThreshold != nil {
+		numEdgeThreshold = *edgeThreshold
+	}
+
+	var numSigma float64 = 1.6
+
+	if sigma != nil {
+		numSigma = *sigma
+	}
+
+	return SIFT{p: unsafe.Pointer(C.SIFT_CreateCustom(C.int(numFeatures), C.int(numOctaveLayers), C.double(numContrastThreshold), C.double(numEdgeThreshold), C.double(numSigma)))}
+}
+
 // Close SIFT.
 func (d *SIFT) Close() error {
 	C.SIFT_Close((C.SIFT)(d.p))

features2d.h

@@ -97,6 +97,7 @@ struct MultiDMatches FlannBasedMatcher_KnnMatch(FlannBasedMatcher f, Mat query,
 void DrawKeyPoints(Mat src, struct KeyPoints kp, Mat dst, const Scalar s, int flags);
 
 SIFT SIFT_Create();
+SIFT SIFT_CreateCustom(int nfeatures, int nOctaveLayers, double contrastThreshold, double edgeThreshold, double sigma);
 void SIFT_Close(SIFT f);
 struct KeyPoints SIFT_Detect(SIFT f, Mat src);
 struct KeyPoints SIFT_Compute(SIFT f, Mat src, struct KeyPoints kp, Mat desc);