Converter: added new color transfer modes: mkl, mkl-m, idt, idt-m

2019-09-07 22:01:01 +04:00
parent 7ed38a8097
commit bef4e5d33c
4 changed files with 109 additions and 93 deletions
@@ -105,8 +105,8 @@ half_face_mask_mode_dict = {1:'learned',
                                    4:'FAN-dst',
                                    7:'learned*FAN-dst'}

-ctm_dict = { 0: "None", 1:"rct", 2:"lct", 3:"ebs" }
-ctm_str_dict = {None:0, "rct":1, "lct": 2, "ebs":3 }
+ctm_dict = { 0: "None", 1:"rct", 2:"lct", 3:"mkl", 4:"mkl-m", 5:"idt", 6:"idt-m", 7:"ebs" }
+ctm_str_dict = {None:0, "rct":1, "lct":2, "mkl":3, "mkl-m":4, "idt":5, "idt-m":6, "ebs":7 }

 class ConverterConfigMasked(ConverterConfig):