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

Maintain same seed across samples

Browse source
This commit is contained in:
jh 2021-03-12 14:32:41 -08:00
commit 4cf5d66865
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
samplelib/SampleProcessor.py
View file

@ -235,7 +235,7 @@ class SampleProcessor(object):
if channel_type == SPCT.BGR: if channel_type == SPCT.BGR:
out_sample = img out_sample = img
elif channel_type == SPCT.LAB_RAND_TRANSFORM: elif channel_type == SPCT.LAB_RAND_TRANSFORM:
out_sample = random_lab_rotation(img) out_sample = random_lab_rotation(img, sample_rnd_seed)
elif channel_type == SPCT.G: elif channel_type == SPCT.G:
out_sample = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)[...,None] out_sample = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)[...,None]
elif channel_type == SPCT.GGG: elif channel_type == SPCT.GGG: