david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bgr_shuffle randomizer

Browse Source
This commit is contained in:
iperov
2019-04-20 18:05:11 +04:00
parent dcbdc79e84
commit 242b850245
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
samplelib/SampleProcessor.py
+2 -1
View File
@@ -211,7 +211,8 @@ class SampleProcessor(object):
if f & SPTF.MODE_BGR != 0: if f & SPTF.MODE_BGR != 0:
img = img_bgr img = img_bgr
elif f & SPTF.MODE_BGR_SHUFFLE != 0: elif f & SPTF.MODE_BGR_SHUFFLE != 0:
img_bgr = np.take (img_bgr, np.random.permutation(img_bgr.shape[-1]), axis=-1) rnd_state = np.random.RandomState (sample_rnd_seed)
img_bgr = np.take (img_bgr, rnd_state.permutation(img_bgr.shape[-1]), axis=-1)
img = np.concatenate ( (img_bgr,img_mask) , -1 ) img = np.concatenate ( (img_bgr,img_mask) , -1 )
elif f & SPTF.MODE_G != 0: elif f & SPTF.MODE_G != 0:
img = np.concatenate ( (np.expand_dims(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY),-1),img_mask) , -1 ) img = np.concatenate ( (np.expand_dims(cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY),-1),img_mask) , -1 )