Merge pull request #20 from iriapantazi/gaussian

add mean in gaussian function
bethgelab · Apr 29, 2022 · c8c1828 · c8c1828
2 parents d4ce97e + c6dc004
commit c8c1828
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imagecorruptions/corruptions.py b/imagecorruptions/corruptions.py
@@ -108,7 +108,7 @@ def getOptimalKernelWidth1D(radius, sigma):
     return radius * 2 + 1
 
 def gauss_function(x, mean, sigma):
-    return (np.exp(- x**2 / (2 * (sigma**2)))) / (np.sqrt(2 * np.pi) * sigma)
+    return (np.exp(- (x - mean)**2 / (2 * (sigma**2)))) / (np.sqrt(2 * np.pi) * sigma)
 
 def getMotionBlurKernel(width, sigma):
     k = gauss_function(np.arange(width), 0, sigma)