add normalization to the model

FaceONNX · Jul 1, 2024 · 874233c · 874233c
1 parent 6d8ffa7
commit 874233c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/netstandard/DepthONNX/DepthEstimator.cs b/netstandard/DepthONNX/DepthEstimator.cs
@@ -125,7 +125,8 @@ public DepthEstimator(SessionOptions options, DepthEstimatorQuality depthSegment
                 }
             }
 
-            return mask.ResizePreserved(height, width, interpolationMode);
+            // normalize and resize
+            return mask.Normalized().ResizePreserved(height, width, interpolationMode);
         }
 
         #endregion

diff --git a/netstandard/Examples/DepthEstimation/Form1.cs b/netstandard/Examples/DepthEstimation/Form1.cs
@@ -43,7 +43,7 @@ private void Process(Bitmap image)
                 image: image,
                 interpolationMode: InterpolationMode.Bicubic);
 
-            var mask = results.Normalized().FromGrayscale();
+            var mask = results.FromGrayscale();
             image?.Dispose();
 
             //mask.Save("output.png", System.Drawing.Imaging.ImageFormat.Png);