Merge pull request #326 from alexlib/master

update filters.py
OpenPIV · Jul 19, 2024 · 402646e · 402646e
2 parents 713586c + bf98c9f
commit 402646e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/openpiv/filters.py b/openpiv/filters.py
@@ -44,8 +44,13 @@ def _gaussian_kernel(half_width: int=1)-> np.ndarray:
 
     """
     # size = int(half_width)
-    x, y = np.mgrid[-half_width:half_width + 1, -half_width:half_width + 1]
-    g = np.exp(-(x ** 2 / float(half_width) + y ** 2 / float(half_width)))
+
+    if half_width == 0:
+        return 1
+    else: 
+        x, y = np.mgrid[-half_width:half_width + 1, -half_width:half_width + 1]
+        g = np.exp(-(x ** 2 / float(half_width) + y ** 2 / float(half_width)))
+
     return g / g.sum()