From c04740eac1cf0e7fc3cb11f49862ac5800fd87c8 Mon Sep 17 00:00:00 2001 From: Colombo Date: Tue, 17 Dec 2019 09:53:17 +0400 Subject: [PATCH] 1 --- samplelib/SampleGeneratorFacePerson.py | 36 +++++++++++++------------- 1 file changed, 18 insertions(+), 18 deletions(-) diff --git a/samplelib/SampleGeneratorFacePerson.py b/samplelib/SampleGeneratorFacePerson.py index 95d929c..6ec4763 100644 --- a/samplelib/SampleGeneratorFacePerson.py +++ b/samplelib/SampleGeneratorFacePerson.py @@ -43,25 +43,25 @@ class SampleGeneratorFacePerson(SampleGeneratorBase): samples[i] = copy.copy(samples[i]) 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) + # + #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)