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

1

Browse source
This commit is contained in:
Colombo 2019-12-17 09:53:17 +04:00
parent 28b948888c
commit c04740eac1
1 changed files with 18 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

36
samplelib/SampleGeneratorFacePerson.py
View file

@ -43,25 +43,25 @@ class SampleGeneratorFacePerson(SampleGeneratorBase):
samples[i] = copy.copy(samples[i]) samples[i] = copy.copy(samples[i])
if person_id_mode==1: if person_id_mode==1:
np.random.shuffle(samples)
new_samples = []
while len(samples) > 0:
for i in range( len(samples)-1, -1, -1):
sample = samples[i]
if len(sample) > 0:
new_samples.append(sample.pop(0))
if len(sample) == 0:
samples.pop(i)
samples = new_samples
#new_samples = []
#for s in samples:
# new_samples += s
#samples = new_samples
#np.random.shuffle(samples) #np.random.shuffle(samples)
#
#new_samples = []
#while len(samples) > 0:
# for i in range( len(samples)-1, -1, -1):
# sample = samples[i]
#
# if len(sample) > 0:
# new_samples.append(sample.pop(0))
#
# if len(sample) == 0:
# samples.pop(i)
# i -= 1
#samples = new_samples
new_samples = []
for s in samples:
new_samples += s
samples = new_samples
np.random.shuffle(samples)
self.samples_len = len(samples) self.samples_len = len(samples)