nodes/EllangoK-ComfyUI-post-processing-nodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change Vignette behaviour on Film Grain node

Browse Source 
Simplified the vignette functionality on the Film Grain node to support floats (step: 0.01).

This allows very subtle vignette.
This commit is contained in:
John Richard Chipps-Harding
2024-02-07 01:57:51 +00:00
parent de7a29b339
commit 10e5d4062d
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
post_processing/film_grain.py
+4 -4
View File
@@ -32,7 +32,7 @@ class FilmGrain:
"default": 0.0,
"min": 0.0,
"max": 10.0,
"step": 1.0
"step": 0.01
}),
},
}
@@ -140,9 +140,9 @@ class FilmGrain:
X, Y = np.meshgrid(x, y)
radius = np.sqrt(X ** 2 + Y ** 2)
# Map vignette strength from 0-10 to 1.800-0.800
mapped_vignette_strength = 1.8 - (vignette_strength - 1) * 0.1
vignette = 1 - np.clip(radius / mapped_vignette_strength, 0, 1)
radius = radius / np.max(radius)
opacity = np.clip(vignette_strength, 0, 1)
vignette = 1 - radius * opacity
return np.clip(image * vignette[..., np.newaxis], 0, 1)
post_processing_nodes.py
+4 -4
View File
@@ -620,7 +620,7 @@ class FilmGrain:
"default": 0.0,
"min": 0.0,
"max": 10.0,
"step": 1.0
"step": 0.01
}),
},
}
@@ -728,9 +728,9 @@ class FilmGrain:
X, Y = np.meshgrid(x, y)
radius = np.sqrt(X ** 2 + Y ** 2)
# Map vignette strength from 0-10 to 1.800-0.800
mapped_vignette_strength = 1.8 - (vignette_strength - 1) * 0.1
vignette = 1 - np.clip(radius / mapped_vignette_strength, 0, 1)
radius = radius / np.max(radius)
opacity = np.clip(vignette_strength, 0, 1)
vignette = 1 - radius * opacity
return np.clip(image * vignette[..., np.newaxis], 0, 1)