fanwuyang/neural-enhance-super-resolution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix histogram colors overflowing in certain images. Closes #73.

Browse Source
This commit is contained in:
Alex J. Champandard
2016-11-12 15:43:23 +01:00
parent 904071d993
commit 9d2aa3c436
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance.py
+1 -1
View File
@@ -542,7 +542,7 @@ class NeuralEnhancer(object):
Hpa, Hpb = [np.cumsum(i) * (rng[1] - rng[0]) ** 2 / float(bins) for i in [Ha, Hb]] Hpa, Hpb = [np.cumsum(i) * (rng[1] - rng[0]) ** 2 / float(bins) for i in [Ha, Hb]]
inv_Ha = scipy.interpolate.interp1d(X, Hpa, bounds_error=False) inv_Ha = scipy.interpolate.interp1d(X, Hpa, bounds_error=False)
map_Hb = scipy.interpolate.interp1d(Hpb, X, bounds_error=False) map_Hb = scipy.interpolate.interp1d(Hpb, X, bounds_error=False)
return map_Hb(inv_Ha(A)) return map_Hb(inv_Ha(A).clip(0.0, 255.0))
def process(self, original): def process(self, original):
# Snap the image to a shape that's compatible with the generator (2x, 4x) # Snap the image to a shape that's compatible with the generator (2x, 4x)