mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2025-01-07 09:54:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

use mask_color variable

Browse Source
This commit is contained in:
deeppomf 2018-02-27 11:36:20 -05:00
parent 092c214103
commit 4e4c2d74ef
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
decensor.py
View File

@ -74,7 +74,7 @@ def get_mask(x_batch):
m = np.zeros((args.image_size, args.image_size, 1), dtype=np.uint8) m = np.zeros((args.image_size, args.image_size, 1), dtype=np.uint8)
for x in range(args.image_size): for x in range(args.image_size):
for y in range(args.image_size): for y in range(args.image_size):
if np.array_equal(raw[x][y], [0, 255, 0]): if np.array_equal(raw[x][y], mask_color):
m[x, y] = 1 m[x, y] = 1
mask.append(m) mask.append(m)
return np.array(mask) return np.array(mask)