From 663af7c1c8bb5aa8ccd6bb1634b51f7ed7a32740 Mon Sep 17 00:00:00 2001 From: Brian Drupieski Date: Sat, 17 Sep 2022 13:57:01 -0400 Subject: [PATCH] fix batch color correction --- scripts/batch.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/scripts/batch.py b/scripts/batch.py index 1af4a7bc3..fc9e6b5fe 100644 --- a/scripts/batch.py +++ b/scripts/batch.py @@ -6,7 +6,7 @@ import traceback import modules.scripts as scripts import gradio as gr -from modules.processing import Processed, process_images +from modules.processing import Processed, process_images, setup_color_correction from PIL import Image from modules.shared import opts, cmd_opts, state @@ -51,6 +51,10 @@ class Script(scripts.Script): state.job = f"{batch_no} out of {batch_count}: {batch_images[0][1]}" p.init_images = [x[0] for x in batch_images] + + if opts.img2img_color_correction: + p.color_corrections = [setup_color_correction(i) for i in p.init_images] + proc = process_images(p) for image, (_, path) in zip(proc.images, batch_images): filename = os.path.basename(path) -- GitLab