From 7bc5739fe27809802ed24d1833b0ef443ec64aa9 Mon Sep 17 00:00:00 2001 From: Brian Drupieski Date: Sat, 17 Sep 2022 18:18:30 -0400 Subject: set loopback color corrections on each iteration --- scripts/batch.py | 5 +---- scripts/loopback.py | 4 ++-- 2 files changed, 3 insertions(+), 6 deletions(-) (limited to 'scripts') diff --git a/scripts/batch.py b/scripts/batch.py index fc9e6b5f..a117fcbf 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, setup_color_correction +from modules.processing import Processed, process_images from PIL import Image from modules.shared import opts, cmd_opts, state @@ -52,9 +52,6 @@ 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) diff --git a/scripts/loopback.py b/scripts/loopback.py index 8aca61f3..2a3f4d6d 100644 --- a/scripts/loopback.py +++ b/scripts/loopback.py @@ -40,8 +40,7 @@ class Script(scripts.Script): all_images = [] state.job_count = loops * batch_count - if opts.img2img_color_correction: - p.color_corrections = [processing.setup_color_correction(p.init_images[0])] + initial_color_corrections = [processing.setup_color_correction(p.init_images[0])] for n in range(batch_count): history = [] @@ -50,6 +49,7 @@ class Script(scripts.Script): p.n_iter = 1 p.batch_size = 1 p.do_not_save_grid = True + p.color_corrections = initial_color_corrections state.job = f"Iteration {i + 1}/{loops}, batch {n + 1}/{batch_count}" -- cgit v1.2.1