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

get rid of random module

Browse source
This commit is contained in:
iperov 2019-03-27 10:28:07 +04:00
parent f63f78e1ab
commit 24dd4ef000
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
samples/SampleGeneratorFace.py
View file

@ -1,6 +1,5 @@
import traceback import traceback
import numpy as np import numpy as np
import random
import cv2 import cv2
import multiprocessing import multiprocessing
from utils import iter_utils from utils import iter_utils
@ -13,7 +12,7 @@ from samples import SampleGeneratorBase
''' '''
arg arg
output_sample_types = [ output_sample_types = [
[SampleProcessor.TypeFlags, size, (optional)random_sub_size] , [SampleProcessor.TypeFlags, size, (optional)random_sub_size] ,
... ...
] ]
@ -119,7 +118,8 @@ class SampleGeneratorFace(SampleGeneratorBase):
idx = shuffle_idxs.pop() idx = shuffle_idxs.pop()
if samples[idx] != None: if samples[idx] != None:
if len(shuffle_idxs_2D[idx]) == 0: if len(shuffle_idxs_2D[idx]) == 0:
shuffle_idxs_2D[idx] = random.sample( range(len(samples[idx])), len(samples[idx]) ) a = shuffle_idxs_2D[idx] = [ *range(len(samples[idx])) ]
np.random.shuffle (a)
idx2 = shuffle_idxs_2D[idx].pop() idx2 = shuffle_idxs_2D[idx].pop()
sample = samples[idx][idx2] sample = samples[idx][idx2]

4
samples/SampleGeneratorImageTemporal.py
View file

@ -1,6 +1,5 @@
import traceback import traceback
import numpy as np import numpy as np
import random
import cv2 import cv2
from utils import iter_utils from utils import iter_utils
@ -58,7 +57,8 @@ class SampleGeneratorImageTemporal(SampleGeneratorBase):
for n_batch in range(self.batch_size): for n_batch in range(self.batch_size):
if len(shuffle_idxs) == 0: if len(shuffle_idxs) == 0:
shuffle_idxs = random.sample( range(samples_sub_len), samples_sub_len ) shuffle_idxs = [ *range(samples_sub_len) ]
np.random.shuffle (shuffle_idxs)
idx = shuffle_idxs.pop() idx = shuffle_idxs.pop()