From 4e4c2d74efa0d18cc9891baab5ace90fc1872dbb Mon Sep 17 00:00:00 2001
From: deeppomf <deeppomf@gmail.com>
Date: Tue, 27 Feb 2018 11:36:20 -0500
Subject: [PATCH] use mask_color variable

---
 decensor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/decensor.py b/decensor.py
index 3526d4a..c7054d0 100644
--- a/decensor.py
+++ b/decensor.py
@@ -74,7 +74,7 @@ def get_mask(x_batch):
         m = np.zeros((args.image_size, args.image_size, 1), dtype=np.uint8)
         for x 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
         mask.append(m)
     return np.array(mask)