From 7066b1122a11bba244777052b24c103e612f7977 Mon Sep 17 00:00:00 2001
From: Daniil Kazantsev <dkazanc@hotmail.com>
Date: Wed, 2 May 2018 11:18:56 +0100
Subject: NVM

---
 Core/inpainters_CPU/NonlocalMarching_Inpaint_core.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Core/inpainters_CPU/NonlocalMarching_Inpaint_core.c b/Core/inpainters_CPU/NonlocalMarching_Inpaint_core.c
index d6cdf93..3a846bb 100644
--- a/Core/inpainters_CPU/NonlocalMarching_Inpaint_core.c
+++ b/Core/inpainters_CPU/NonlocalMarching_Inpaint_core.c
@@ -56,7 +56,9 @@ float NonlocalMarching_Inpaint_main(float *Input, unsigned char *M, float *Outpu
             if (M[i] == 1) iterations_number++;
         }
     }
-    else iterations_number = iterationsNumb;
+    else iterations_number = (int)(iterationsNumb/dimX);
+    if (iterations_number > dimX) iterations_number = dimX;
+    
     if (iterations_number == 0) printf("%s \n", "Nothing to inpaint, zero mask!");
     else {
         
-- 
cgit v1.2.3