Fix another 2 issues causing ocr crash

sdcb · Oct 27, 2023 · 15c180d · 15c180d
1 parent 04811c4
commit 15c180d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/projects/PaddleOCR/Sdcb.OpenVINO.PaddleOCR/PaddleOcrClassifier.cs b/projects/PaddleOCR/Sdcb.OpenVINO.PaddleOCR/PaddleOcrClassifier.cs
@@ -101,7 +101,7 @@ public Ocr180DegreeClsResult[] ShouldRotate180(Mat[] srcs)
 
     private Ocr180DegreeClsResult[] BatchedShouldRotate180(Mat[] srcs)
     {
-        Mat final = PrepareAndStackImages(srcs);
+        using Mat final = PrepareAndStackImages(srcs);
 
         using (Tensor input = final.StackedAsTensor(srcs.Length))
         {

diff --git a/projects/PaddleOCR/Sdcb.OpenVINO.PaddleOCR/PaddleOcrRecognizer.cs b/projects/PaddleOCR/Sdcb.OpenVINO.PaddleOCR/PaddleOcrRecognizer.cs
@@ -135,7 +135,7 @@ private unsafe PaddleOcrRecognizerResult[] BatchedRun(Mat[] srcs)
             return padded;
         }));
 
-        Mat final = PrepareAndStackImages(srcs, modelHeight, maxWidth);
+        using Mat final = PrepareAndStackImages(srcs, modelHeight, maxWidth);
 
         using (Tensor input = final.StackedAsTensor(srcs.Length))
         {