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

Adds masking to Sample Processor

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-11 16:28:11 -07:00
commit 2caa70ca00
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
imagelib/color_transfer.py
View file

@ -51,8 +51,8 @@ def reinhard_color_transfer(target, source, clip=False, preserve_paper=False, ta
target = cv2.cvtColor(target.astype(np.float32), cv2.COLOR_BGR2LAB) target = cv2.cvtColor(target.astype(np.float32), cv2.COLOR_BGR2LAB)
# compute color statistics for the source and target images # compute color statistics for the source and target images
(lMeanSrc, lStdSrc, aMeanSrc, aStdSrc, bMeanSrc, bStdSrc) = lab_image_stats(src_input) (lMeanSrc, lStdSrc, aMeanSrc, aStdSrc, bMeanSrc, bStdSrc) = lab_image_stats(source, mask=source_mask)
(lMeanTar, lStdTar, aMeanTar, aStdTar, bMeanTar, bStdTar) = lab_image_stats(tgt_input) (lMeanTar, lStdTar, aMeanTar, aStdTar, bMeanTar, bStdTar) = lab_image_stats(target, mask=target_mask)
# subtract the means from the target image # subtract the means from the target image
(l, a, b) = cv2.split(target) (l, a, b) = cv2.split(target)