Reset init img in loopback at start of each batch (#2214)

Before a new batch would use the last image from the previous batch. Now
each batch will use the original image for the init image at the start of the
batch.
This commit is contained in:
Rory Grieve 2022-10-11 11:23:47 +01:00 committed by GitHub
parent 255be75d30
commit 4b460fcb1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions

View File

@ -38,6 +38,7 @@ class Script(scripts.Script):
grids = []
all_images = []
original_init_image = p.init_images
state.job_count = loops * batch_count
initial_color_corrections = [processing.setup_color_correction(p.init_images[0])]
@ -45,6 +46,9 @@ class Script(scripts.Script):
for n in range(batch_count):
history = []
# Reset to original init image at the start of each batch
p.init_images = original_init_image
for i in range(loops):
p.n_iter = 1
p.batch_size = 1