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

fix shuffle

Browse source
This commit is contained in:
Jeremy Hummel 2021-05-23 22:32:43 -07:00
commit 924c3fc6fd
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
samplelib/SampleProcessor.py
View file

@ -217,7 +217,9 @@ class SampleProcessor(object):
ct_sample_bgr = ct_sample.load_bgr() ct_sample_bgr = ct_sample.load_bgr()
img = imagelib.color_transfer (ct_mode, img, cv2.resize( ct_sample_bgr, (resolution,resolution), interpolation=cv2.INTER_LINEAR ) ) img = imagelib.color_transfer (ct_mode, img, cv2.resize( ct_sample_bgr, (resolution,resolution), interpolation=cv2.INTER_LINEAR ) )
randomization_order = np.random.shuffle(['blur', 'noise', 'jpeg', 'down'])
randomization_order = ['blur', 'noise', 'jpeg', 'down']
np.random.shuffle(randomization_order)
for random_distortion in randomization_order: for random_distortion in randomization_order:
# Apply random blur # Apply random blur
if random_distortion == 'blur' and random_blur: if random_distortion == 'blur' and random_blur: