aboutsummaryrefslogtreecommitdiff
path: root/scripts/outpainting_mk_2.py
diff options
context:
space:
mode:
authorJocke <joakim.lovgren@protonmail.com>2022-10-05 16:31:48 +0200
committerAUTOMATIC1111 <16777216c@gmail.com>2022-10-06 11:23:47 +0300
commita93c3ffbfd264ed6b5d989922352300c9d3efbe4 (patch)
tree0918fcea6f60da7789b1b06aaa7576b7ee5568fd /scripts/outpainting_mk_2.py
parent4288e53fc2ea25fa49715bf5b7f14603553c9e38 (diff)
Outpainting mk2, prevent generation of a completely random image every time even when global seed is static
Diffstat (limited to 'scripts/outpainting_mk_2.py')
-rw-r--r--scripts/outpainting_mk_2.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/scripts/outpainting_mk_2.py b/scripts/outpainting_mk_2.py
index 11613ca3..a6468e09 100644
--- a/scripts/outpainting_mk_2.py
+++ b/scripts/outpainting_mk_2.py
@@ -85,8 +85,11 @@ def get_matched_noise(_np_src_image, np_mask_rgb, noise_q=1, color_variation=0.0
src_dist = np.absolute(src_fft)
src_phase = src_fft / src_dist
+ # create a generator with a static seed to make outpainting deterministic / only follow global seed
+ rng = np.random.default_rng(0)
+
noise_window = _get_gaussian_window(width, height, mode=1) # start with simple gaussian noise
- noise_rgb = np.random.random_sample((width, height, num_channels))
+ noise_rgb = rng.random((width, height, num_channels))
noise_grey = (np.sum(noise_rgb, axis=2) / 3.)
noise_rgb *= color_variation # the colorfulness of the starting noise is blended to greyscale with a parameter
for c in range(num_channels):