From bf094c31876dc418ac0b4cf756d4b37be8af8a11 Mon Sep 17 00:00:00 2001 From: vikasTmz Date: Thu, 20 Dec 2018 19:03:47 +0100 Subject: [PATCH] LCN function fix --- LocalContrastNorm.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/LocalContrastNorm.ipynb b/LocalContrastNorm.ipynb index 05cb5ca..d79bb75 100644 --- a/LocalContrastNorm.ipynb +++ b/LocalContrastNorm.ipynb @@ -639,7 +639,7 @@ " ## Variance Calc\n", " sum_sqr_image = F.conv2d(centered_image.pow(2),gaussian_filter,padding=radius-1)\n", " s_deviation = sum_sqr_image[:,:,mid:-mid,mid:-mid].sqrt()\n", - " per_img_mean = denom.mean()\n", + " per_img_mean = s_deviation.mean()\n", " \n", " ## Divisive Normalization\n", " divisor = np.maximum(per_img_mean.numpy(),s_deviation.numpy())\n",