was-node-suite-comfyui icon indicating copy to clipboard operation
was-node-suite-comfyui copied to clipboard
verified publisher icon WASasquatch

Use GPU for otherwise very slow "Image Film Grain" node (if available)

Open ClownsharkBatwing opened this issue 1 year ago • 2 comments

The "Image Film Grain" node was effectively generating large noised images one pixel at a time looping through randint() calls, which is a very slow process. This forked version uses pytorch and is much faster - on a 5950x + 4090 system, 1024x1024 images went from taking at least 30 seconds to maybe 0.2 seconds. The benefits are even greater with larger images, which drop from taking several minutes to under a second.

ClownsharkBatwing avatar Aug 15 '24 17:08 ClownsharkBatwing

Oh hey thanks. Yeah a lot of the noise/image stuff could be moved to torch. Will most. Some stuff I found PyTorch implementations of kinda inferior. Like doing sharpening.

WASasquatch avatar Aug 15 '24 21:08 WASasquatch

Did you mean to include a "CLIPSeg2" in here?

WASasquatch avatar Oct 09 '24 23:10 WASasquatch