Update process

sbrunner · Apr 12, 2019 · 7edab2f · 7edab2f
1 parent 92b92ad
commit 7edab2f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/process b/process
@@ -101,9 +101,9 @@ class Context:  # pylint: disable=too-many-instance-attributes
         if self.mask_ready is None:
             return self.image
 
-        white = np.zeros(self.image.shape)
-        white[:] = (255, 255, 255)
-        return np.where(self.mask == 0, white, self.image)
+        image = self.image.copy()
+        image[self.mask_ready == 0] = (255, 255, 255)
+        return image
 
     def crop(self, x: int, y: int, width: int, height: int) -> None:
         if self.image is None: