Expose FaceDetectorYN.SetInputSize

Author: shimat

src/OpenCvSharp/Internal/PInvoke/NativeMethods/objdetect/NativeMethods_objdetect_FaceDetectorYN.cs

@@ -28,6 +28,10 @@ public static partial ExceptionStatus objdetect_FaceDetectorYN_create(
     [LibraryImport(DllExtern), UnmanagedCallConv(CallConvs = [typeof(CallConvCdecl)])]
     public static partial ExceptionStatus objdetect_Ptr_FaceDetectorYN_get(IntPtr ptr, out IntPtr returnValue);
 
+    [LibraryImport(DllExtern), UnmanagedCallConv(CallConvs = [typeof(CallConvCdecl)])]
+    internal static partial ExceptionStatus objdetect_FaceDetectorYN_setInputSize(
+        OpenCvSafeHandle obj, Size inputSize);
+
     [LibraryImport(DllExtern), UnmanagedCallConv(CallConvs = [typeof(CallConvCdecl)])]
     internal static partial ExceptionStatus objdetect_FaceDetectorYN_detect(
         OpenCvSafeHandle obj, in InputArrayProxy image, in OutputArrayProxy faces, out int returnValue);

src/OpenCvSharp/Modules/objdetect/FaceDetectorYN.cs

@@ -56,6 +56,24 @@ public static FaceDetectorYN Create(
         return new FaceDetectorYN(smartPtr, rawPtr);
     }
 
+    /// <summary>
+    /// Sets the network input size, overwriting the size specified when the detector was created.
+    /// </summary>
+    /// <param name="inputSize">The size of the input image.</param>
+    public void SetInputSize(Size inputSize)
+    {
+        ThrowIfDisposed();
+        NativeMethods.HandleException(
+            NativeMethods.objdetect_FaceDetectorYN_setInputSize(Handle, inputSize));
+    }
+
+    /// <summary>
+    /// Sets the network input size, overwriting the size specified when the detector was created.
+    /// </summary>
+    /// <param name="width">The input image width.</param>
+    /// <param name="height">The input image height.</param>
+    public void SetInputSize(int width, int height) => SetInputSize(new Size(width, height));
+
     /// <summary>
     /// A simple interface to detect face from given image.
     /// </summary>

src/OpenCvSharpExtern/objdetect_FaceDetectorYN.h

@@ -46,6 +46,15 @@ CVAPI(ExceptionStatus) objdetect_Ptr_FaceDetectorYN_get(cv::Ptr<cv::FaceDetector
     });
 }
 
+CVAPI(ExceptionStatus) objdetect_FaceDetectorYN_setInputSize(
+    cv::FaceDetectorYN* obj,
+    const interop::Size inputSize)
+{
+    return cvTry([&] {
+        obj->setInputSize(cpp(inputSize));
+    });
+}
+
 CVAPI(ExceptionStatus) objdetect_FaceDetectorYN_detect(
     cv::FaceDetectorYN* obj,
     const interop::InputArrayProxy* image,

test/OpenCvSharp.Tests/objdetect/FaceDetectorYNTest.cs

@@ -49,6 +49,22 @@ public void CreateWithParameters()
         Assert.NotNull(detector);
     }
 
+    [Fact]
+    public void SetInputSize()
+    {
+        using var image = LoadImage("lenna.png");
+        using var detector = FaceDetectorYN.Create(
+            ModelPath,
+            config: "",
+            inputSize: new Size(320, 320));
+
+        detector.SetInputSize(image.Width, image.Height);
+
+        using var faces = new Mat();
+        Assert.Equal(1, detector.Detect(image, faces));
+        Assert.False(faces.Empty());
+    }
+
     [Fact]
     public void DisposeTwice()
     {