CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

revert for more tests

Browse Source
This commit is contained in:
iperov
2021-04-27 19:22:11 +04:00
parent e53d1b1820
commit 87030bdcdf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samplelib/SampleGeneratorFaceXSeg.py
+2 -2
View File
@@ -154,8 +154,8 @@ class SampleGeneratorFaceXSeg(SampleGeneratorBase):
mask[mask >= 0.5] = 1.0 mask[mask >= 0.5] = 1.0
mask = np.clip(mask, 0, 1) mask = np.clip(mask, 0, 1)
if np.random.randint(4) < 3: #if np.random.randint(4) < 3:
img = imagelib.apply_random_relight(img) # img = imagelib.apply_random_relight(img)
if np.random.randint(2) == 0: if np.random.randint(2) == 0:
img = imagelib.apply_random_hsv_shift(img, mask=sd.random_circle_faded ([resolution,resolution])) img = imagelib.apply_random_hsv_shift(img, mask=sd.random_circle_faded ([resolution,resolution]))