From 5f811a1b758435929883e922ce7deaa533aa9cba Mon Sep 17 00:00:00 2001 From: iperov Date: Sun, 31 Mar 2019 16:48:36 +0400 Subject: [PATCH] remove asking color transfer in raw mode --- converters/ConverterMasked.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/converters/ConverterMasked.py b/converters/ConverterMasked.py index dff0e42..f599625 100644 --- a/converters/ConverterMasked.py +++ b/converters/ConverterMasked.py @@ -89,7 +89,10 @@ class ConverterMasked(Converter): self.blur_mask_modifier = base_blur_mask_modifier + np.clip ( io.input_int ("Choose blur mask modifier [-200..200] (skip:%d) : " % (default_blur_mask_modifier), default_blur_mask_modifier), -200, 200) self.output_face_scale = np.clip ( 1.0 + io.input_int ("Choose output face scale modifier [-50..50] (skip:0) : ", 0)*0.01, 0.5, 1.5) - self.color_transfer_mode = io.input_str ("Apply color transfer to predicted face? Choose mode ( rct/lct skip:None ) : ", None, ['rct','lct']) + + if self.mode != 'raw': + self.color_transfer_mode = io.input_str ("Apply color transfer to predicted face? Choose mode ( rct/lct skip:None ) : ", None, ['rct','lct']) + self.super_resolution = io.input_bool("Apply super resolution? (y/n ?:help skip:n) : ", False, help_message="Enhance details by applying DCSCN network.") if self.mode != 'raw':