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.

1 files changed, 4 insertions, 0 deletions

diff --git a/scripts/loopback.py b/scripts/loopback.py
index e90b58d4..d8c68af8 100644
--- a/scripts/loopback.py
+++ b/scripts/loopback.py
@@ -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