From fe5b659f10be69ac887f9ad224db4d02d4bd1a93 Mon Sep 17 00:00:00 2001 From: chflame163 Date: Sun, 15 Dec 2024 10:19:51 +0800 Subject: [PATCH] fix noise random of Film nodes --- py/film_post.py | 6 +++--- py/film_post_v2.py | 6 ++++-- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/py/film_post.py b/py/film_post.py index 4652b32..2fdca4a 100644 --- a/py/film_post.py +++ b/py/film_post.py @@ -47,7 +47,7 @@ class Film: ): ret_images = [] - + seed = int(time.time()) for i in image: i = torch.unsqueeze(i, 0) _canvas = tensor2pil(i).convert('RGB') @@ -74,8 +74,8 @@ class Film: _canvas = vignette_image(_canvas, vignette_intensity, center_x, center_y) if grain_power: - _canvas = filmgrain_image(_canvas, grain_scale, grain_power, grain_shadows, grain_highs, grain_sat, seed=int(time.time())) - + _canvas = filmgrain_image(_canvas, grain_scale, grain_power, grain_shadows, grain_highs, grain_sat, seed=seed) + seed += 1 ret_image = _canvas ret_images.append(pil2tensor(ret_image)) diff --git a/py/film_post_v2.py b/py/film_post_v2.py index 6efe3c9..de1e0ef 100644 --- a/py/film_post_v2.py +++ b/py/film_post_v2.py @@ -48,6 +48,7 @@ class FilmV2: ): ret_images = [] + seed = int(time.time()) for i in image: i = torch.unsqueeze(i, 0) @@ -76,9 +77,10 @@ class FilmV2: if grain_power: if grain_method == "fastgrain": - _canvas = image_add_grain(_canvas, grain_scale,grain_power, grain_sat, toe=0, seed=int(time.time())) + _canvas = image_add_grain(_canvas, grain_scale,grain_power, grain_sat, toe=0, seed=seed) elif grain_method == "filmgrainer": - _canvas = filmgrain_image(_canvas, grain_scale, grain_power, filmgrainer_shadows, filmgrainer_highs, grain_sat, seed=int(time.time())) + _canvas = filmgrain_image(_canvas, grain_scale, grain_power, filmgrainer_shadows, filmgrainer_highs, grain_sat, seed=seed) + seed += 1 ret_image = _canvas ret_images.append(pil2tensor(ret_image))