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

SampleGeneratorFaceXSeg: additional sample augmentation for better generalization

Browse Source
This commit is contained in:
iperov
2021-05-12 09:42:29 +04:00
parent 8b90ca0dac
commit 65a703c024
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samplelib/SampleGeneratorFaceXSeg.py
+11 -5
View File
@@ -151,11 +151,6 @@ class SampleGeneratorFaceXSeg(SampleGeneratorBase):
mask[mask < 0.5] = 0.0 mask[mask < 0.5] = 0.0
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(2) == 0:
img = imagelib.apply_random_hsv_shift(img, mask=sd.random_circle_faded ([resolution,resolution]))
else:
img = imagelib.apply_random_rgb_levels(img, mask=sd.random_circle_faded ([resolution,resolution]))
if np.random.randint(2) == 0: if np.random.randint(2) == 0:
# random face flare # random face flare
@@ -163,6 +158,17 @@ class SampleGeneratorFaceXSeg(SampleGeneratorBase):
krn = krn - krn % 2 + 1 krn = krn - krn % 2 + 1
img = img + cv2.GaussianBlur(img*mask, (krn,krn), 0) img = img + cv2.GaussianBlur(img*mask, (krn,krn), 0)
if np.random.randint(2) == 0:
# random bg flare
krn = np.random.randint( resolution//4, resolution )
krn = krn - krn % 2 + 1
img = img + cv2.GaussianBlur(img*(1-mask), (krn,krn), 0)
if np.random.randint(2) == 0:
img = imagelib.apply_random_hsv_shift(img, mask=sd.random_circle_faded ([resolution,resolution]))
else:
img = imagelib.apply_random_rgb_levels(img, mask=sd.random_circle_faded ([resolution,resolution]))
if np.random.randint(2) == 0: if np.random.randint(2) == 0:
img = imagelib.apply_random_sharpen( img, sharpen_chance, sharpen_kernel_max_size, mask=sd.random_circle_faded ([resolution,resolution])) img = imagelib.apply_random_sharpen( img, sharpen_chance, sharpen_kernel_max_size, mask=sd.random_circle_faded ([resolution,resolution]))
else: else: