From 127be44f9d6b384e874a73ba794a7eeb8f2b6a0f Mon Sep 17 00:00:00 2001 From: chflame163 Date: Sun, 26 Jan 2025 16:56:10 +0800 Subject: [PATCH] fix add_grain.py for ramdom grain in batch images --- py/add_grain.py | 6 +++--- pyproject.toml | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/py/add_grain.py b/py/add_grain.py index 45f3e8a..b7d4c10 100644 --- a/py/add_grain.py +++ b/py/add_grain.py @@ -33,9 +33,9 @@ class AddGrain: ret_images = [] - for i in image: - _canvas = tensor2pil(torch.unsqueeze(i, 0)).convert('RGB') - _canvas = image_add_grain(_canvas, grain_scale, grain_power, grain_sat, toe=0, seed=int(time.time())) + for i in range(len(image)): + _canvas = tensor2pil(torch.unsqueeze(image[i], 0)).convert('RGB') + _canvas = image_add_grain(_canvas, grain_scale, grain_power, grain_sat, toe=0, seed=int(time.time()) + i) ret_images.append(pil2tensor(_canvas)) log(f"{self.NODE_NAME} Processed {len(ret_images)} image(s).", message_type='finish') diff --git a/pyproject.toml b/pyproject.toml index 9da3493..a7b1442 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui_layerstyle" description = "A set of nodes for ComfyUI it generate image like Adobe Photoshop's Layer Style. the Drop Shadow is first completed node, and follow-up work is in progress." -version = "2.0.12" +version = "2.0.13" license = "MIT" dependencies = ["numpy", "pillow", "torch", "matplotlib", "Scipy", "scikit_image", "scikit_learn", "opencv-contrib-python", "pymatting", "timm", "colour-science", "transformers", "blend_modes", "huggingface_hub", "loguru"]