selection

function.py

@@ -65,9 +65,7 @@ def doTheInpainting(img,mask,radius):
         source = pyramid[level]
         if (level == maxLevel-1):
             target = source.copy()
-            for y in range(target.height):
-                for x in range(target.width):
-                    target.mask[y,x] = False
+            target.mask[0:target.height,0:target.width] = False
             sourceToTarget = Nnf(source,target,radius)
             sourceToTarget.randomize()
 

nnf.py

@@ -10,9 +10,9 @@ class Nnf:
 
     def randomize(self):
         self.field = np.zeros((self.input.height, self.input.width, 3), dtype=int)
-        self.field[:,:,2] += MaskedImage.DSCALE
-        self.field[:,:,0] = np.random.randint(0,self.output.width,self.input.width)
-        self.field[:,:,1] = np.random.randint(0,self.output.height,self.input.height)
+        self.field[:,:,2] = MaskedImage.DSCALE
+        self.field[:,:,0] = np.random.randint(0,self.output.width,(self.input.height,self.input.width))
+        self.field[:,:,1] = np.random.randint(0,self.output.height,(self.input.height,self.input.width))
         self.initialize()
     
     def initializeFromNnf(self,nnf):