fix model inpaint/binary logic

profjsb · Oct 1, 2019 · affd13d · affd13d
1 parent a68d766
commit affd13d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/deepCR/model.py b/deepCR/model.py
@@ -107,6 +107,9 @@ def clean(self, img0, threshold=0.5, inpaint=True, binary=True, segment=False,
         """
 
         # data pre-processing
+
+        inpaint = inpaint and binary
+
         img0 = img0.astype(np.float32) / self.scale
         img0 = img0.copy()
         if self.norm: