From a1e30d2ea03d961e8c0be5b6552591c05d0032ca Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Wed, 26 Feb 2020 13:24:20 +0100
Subject: [PATCH] Implemented in the scoring step of the Vanilla-pipeline
---
bob/bio/base/script/vanilla_biometrics.py | 22 +++++++++++++++-------
1 file changed, 15 insertions(+), 7 deletions(-)
diff --git a/bob/bio/base/script/vanilla_biometrics.py b/bob/bio/base/script/vanilla_biometrics.py
index ad0be863..aa23fd2e 100644
--- a/bob/bio/base/script/vanilla_biometrics.py
+++ b/bob/bio/base/script/vanilla_biometrics.py
@@ -209,18 +209,26 @@ def vanilla_biometrics(
from bob.bio.base.pipelines.vanilla_biometrics.annotated_legacy import SampleLoaderAnnotated as SampleLoader
loader = SampleLoader(pipeline)
- counter = 0
- for g in group:
+ for g in group:
+
with open(os.path.join(output,f"scores-{g}"), "w") as f:
- for biometric_reference in database.references(group=g):
- subject = biometric_reference.subject
- print(f" BIOMETRIC REFERENCE {counter} - {subject}")
- counter += 1
+ # Spliting the references in small chunks
+ n_workers = 3
+ biometric_references = database.references(group=g)
+ offset = 0
+ step = len(biometric_references)//n_workers
+ biometric_references_chunks = []
+ for i in range(n_workers-1):
+ biometric_references_chunks.append(biometric_references[offset:offset+step])
+ offset += step
+ biometric_references_chunks.append(biometric_references[offset:])
+
+ for biometric_reference in biometric_references_chunks:
result = biometric_pipeline(
database.background_model_samples(),
- [biometric_reference],
+ biometric_reference,
database.probes(group=g),
loader,
algorithm,
--
GitLab