github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-14 00:53:48 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

facerelighther: added intensity param

Browse source
This commit is contained in:
Colombo 2019-11-13 00:19:26 +04:00
parent 54786848bc
commit 006233656b
2 changed files with 38 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

10
nnlib/DeepPortraitRelighting.py
View file

@ -45,11 +45,12 @@ class DeepPortraitRelighting(object):
return sh_basis
#n = [0..8]
def relight(self, img, alt, azi, lighten=False):
def relight(self, img, alt, azi, intensity=1.0, lighten=False):
torch = self.torch
sh = self.SH_basis (alt, azi)
sh = (sh.reshape( (1,9,1,1) ) ).astype(np.float32)
sh = (sh.reshape( (1,9,1,1) ) ).astype(np.float32)
#sh *= 0.1
sh = torch.autograd.Variable(torch.from_numpy(sh).to(self.torch_device))
row, col, _ = img.shape
@ -67,9 +68,10 @@ class DeepPortraitRelighting(object):
outputImg = cv2.blur(outputImg, (3,3) )
if not lighten:
outputImg = inputL* outputImg
outputImg = inputL*(1.0-intensity) + (inputL*outputImg)*intensity
else:
outputImg = outputImg*255.0
outputImg = inputL*(1.0-intensity) + (outputImg*255.0)*intensity
outputImg = np.clip(outputImg, 0,255).astype(np.uint8)
Lab[:,:,0] = outputImg