github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 22:34:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

safe division

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-14 12:48:57 -07:00
commit b6368061a6
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
imagelib/color_transfer.py
View file

@ -63,14 +63,14 @@ def reinhard_color_transfer(source, target, clip=False, preserve_paper=False, so
if preserve_paper: if preserve_paper:
# scale by the standard deviations using paper proposed factor # scale by the standard deviations using paper proposed factor
l = (lStdTar / lStdSrc) * l l = (lStdTar / lStdSrc) * l if lStdSrc != 0 else l
a = (aStdTar / aStdSrc) * a a = (aStdTar / aStdSrc) * a if aStdSrc != 0 else l
b = (bStdTar / bStdSrc) * b b = (bStdTar / bStdSrc) * b if bStdSrc != 0 else l
else: else:
# scale by the standard deviations using reciprocal of paper proposed factor # scale by the standard deviations using reciprocal of paper proposed factor
l = (lStdSrc / lStdTar) * l l = (lStdSrc / lStdTar) * l if lStdTar != 0 else l
a = (aStdSrc / aStdTar) * a a = (aStdSrc / aStdTar) * a if aStdTar != 0 else l
b = (bStdSrc / bStdTar) * b b = (bStdSrc / bStdTar) * b if bStdTar != 0 else l
# add in the source mean # add in the source mean
l += lMeanTar l += lMeanTar