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

Merge pull request #6 from digitaljohn/fix/vignette

Browse Source 
Change Vignette behaviour on Film Grain node
This commit is contained in:
Karun
2024-08-08 20:14:29 -04:00
committed by GitHub
parent 2507b7a512 10e5d4062d
commit d5bacf7925
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, "default": 0.0,
"min": 0.0, "min": 0.0,
"max": 10.0, "max": 10.0,
"step": 1.0 "step": 0.01
}), }),
}, },
} }
@@ -140,9 +140,9 @@ class FilmGrain:
X, Y = np.meshgrid(x, y) X, Y = np.meshgrid(x, y)
radius = np.sqrt(X ** 2 + Y ** 2) radius = np.sqrt(X ** 2 + Y ** 2)
# Map vignette strength from 0-10 to 1.800-0.800 radius = radius / np.max(radius)
mapped_vignette_strength = 1.8 - (vignette_strength - 1) * 0.1 opacity = np.clip(vignette_strength, 0, 1)
vignette = 1 - np.clip(radius / mapped_vignette_strength, 0, 1) vignette = 1 - radius * opacity
return np.clip(image * vignette[..., np.newaxis], 0, 1) 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, "default": 0.0,
"min": 0.0, "min": 0.0,
"max": 10.0, "max": 10.0,
"step": 1.0 "step": 0.01
}), }),
}, },
} }
@@ -728,9 +728,9 @@ class FilmGrain:
X, Y = np.meshgrid(x, y) X, Y = np.meshgrid(x, y)
radius = np.sqrt(X ** 2 + Y ** 2) radius = np.sqrt(X ** 2 + Y ** 2)
# Map vignette strength from 0-10 to 1.800-0.800 radius = radius / np.max(radius)
mapped_vignette_strength = 1.8 - (vignette_strength - 1) * 0.1 opacity = np.clip(vignette_strength, 0, 1)
vignette = 1 - np.clip(radius / mapped_vignette_strength, 0, 1) vignette = 1 - radius * opacity
return np.clip(image * vignette[..., np.newaxis], 0, 1) return np.clip(image * vignette[..., np.newaxis], 0, 1)