diff --git a/comfy_extras/nodes_convert_color_space.py b/comfy_extras/nodes_convert_color_space.py index 3f8a23718..f8c7b5568 100644 --- a/comfy_extras/nodes_convert_color_space.py +++ b/comfy_extras/nodes_convert_color_space.py @@ -42,8 +42,8 @@ class ConvertColorSpace(IO.ComfyNode): category="image/color", inputs=[ IO.Image.Input("images"), - IO.Combo.Input("source_color_space", options=["sRGB", "Linear", "HDR (Rec.2020)", "Grayscale"], default="sRGB"), - IO.Combo.Input("target_color_space", options=["sRGB", "Linear", "HDR (Rec.2020)", "Grayscale"], default="Linear"), + IO.Combo.Input("source_color_space", options=["sRGB", "Linear", "HDR Display (PQ/Rec.2020)", "Grayscale"], default="sRGB"), + IO.Combo.Input("target_color_space", options=["sRGB", "Linear", "HDR Display (PQ/Rec.2020)", "Grayscale"], default="Linear"), ], outputs=[ IO.Image.Output("images"), @@ -69,7 +69,7 @@ class ConvertColorSpace(IO.ComfyNode): rgb = luma.unsqueeze(-1).repeat(1, 1, 1, 3) rgb = srgb_to_linear(rgb) - elif source_color_space == "HDR (Rec.2020)": + elif source_color_space == "HDR Display (PQ/Rec.2020)": # assuming Linear Rec.2020 input. Convert to Linear Rec.709 matrix = M_2020_to_709.to(device) rgb = pq_to_linear(rgb) @@ -85,7 +85,7 @@ class ConvertColorSpace(IO.ComfyNode): rgb = luma.unsqueeze(-1).repeat(1, 1, 1, 3) rgb = linear_to_srgb(rgb) # reapply srgb gamma - elif target_color_space == "HDR (Rec.2020)": + elif target_color_space == "HDR Display (PQ/Rec.2020)": # convert Gamut from Linear Rec.709 to Linear Rec.2020 rgb = torch.matmul(rgb, M_709_to_2020.to(device).T).clamp(min=0) rgb = linear_to_pq(rgb) diff --git a/comfy_extras/nodes_images.py b/comfy_extras/nodes_images.py index b7ad14319..373e54e89 100644 --- a/comfy_extras/nodes_images.py +++ b/comfy_extras/nodes_images.py @@ -963,12 +963,6 @@ class SaveImageAdvanced(IO.ComfyNode): default="ComfyUI", tooltip="The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.", ), - IO.Combo.Input( - "interept_as", - options=["Raw/Data", "sRGB"], - default="sRGB", - advanced=True, - ), IO.DynamicCombo.Input( "format", options=[ @@ -981,6 +975,12 @@ class SaveImageAdvanced(IO.ComfyNode): default="8-bit", advanced=True, ), + IO.Combo.Input( + "interept_as", + options=["sRGB", "Linear", "Raw/Data"], + default="sRGB", + advanced=True, + ), ], ), IO.DynamicCombo.Option( @@ -992,6 +992,12 @@ class SaveImageAdvanced(IO.ComfyNode): default="8-bit", advanced=True, ), + IO.Combo.Input( + "interept_as", + options=["sRGB", "Raw/Data"], + default="sRGB", + advanced=True, + ), ], ), IO.DynamicCombo.Option( @@ -1003,6 +1009,12 @@ class SaveImageAdvanced(IO.ComfyNode): default="16-bit", advanced=True, ), + IO.Combo.Input( + "interept_as", + options=["Linear", "Raw/Data"], + default="Linear", + advanced=True, + ), ], ), ], @@ -1025,7 +1037,7 @@ class SaveImageAdvanced(IO.ComfyNode): # get widget values from dynamic combo file_format = format["format"] bit_depth = format["bit_depth"] - color_space = format["color_space"] + interept_as = format["interept_as"] img_tensor = image.clone()