remove xrange for python 3 compat

StartleStars · Feb 12, 2018 · 8f7e46f · 8f7e46f
1 parent 181e13e
commit 8f7e46f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/decensor.py b/decensor.py
@@ -75,8 +75,8 @@ def get_mask(x_batch):
         raw = x_batch[i]
         raw = np.array((raw + 1) * 127.5, dtype=np.uint8)
         m = np.zeros((IMAGE_SIZE, IMAGE_SIZE, 1), dtype=np.uint8)
-        for x in xrange(IMAGE_SIZE):
-            for y in xrange(IMAGE_SIZE):
+        for x in range(IMAGE_SIZE):
+            for y in range(IMAGE_SIZE):
                 if np.array_equal(raw[x][y], [0, 255, 0]):
                     m[x, y] = 1
         mask.append(m)