mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2025-02-22 06:41:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove xrange for python 3 compat

Browse Source
This commit is contained in:
deep pomf 2018-02-12 12:25:10 -05:00
parent 181e13ef68
commit 8f7e46f5f0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
decensor.py
View File

@ -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)