Automatic masks range selection for better visualization (#20)

* Update app.py The range of the mask needs to be changed. * Automatic masks range selection for better visualization Co-authored-by: Konstantin Sofiiuk <[email protected]>
SamsungLabs · Jul 17, 2020 · 90717d6 · 90717d6
1 parent c3f5790
commit 90717d6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/interactive_demo/app.py b/interactive_demo/app.py
@@ -208,6 +208,8 @@ def _save_mask_callback(self):
             mask = self.controller.result_mask
             if mask is None:
                 return
+            if 0 < mask.max() < 256:
+                mask *= 255 // mask.max()
 
             filename = filedialog.asksaveasfilename(parent=self.master, initialfile='{}.png'.format(self.filename), filetypes=[
                 ("PNG image", "*.png"),