Skip to content
Snippets Groups Projects
Commit 9a03e237 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira Committed by Amir MOHAMMADI
Browse files

Updated mobio example

parent cd0700ae
No related branches found
No related tags found
1 merge request!180[dask] Preparing bob.bio.base for dask pipelines
...@@ -8,6 +8,9 @@ import bob.bio.face ...@@ -8,6 +8,9 @@ import bob.bio.face
from bob.bio.base.mixins.legacy import LegacyProcessorMixin, LegacyAlgorithmMixin from bob.bio.base.mixins.legacy import LegacyProcessorMixin, LegacyAlgorithmMixin
from bob.bio.base.pipelines.vanilla_biometrics.legacy import LegacyBiometricAlgorithm from bob.bio.base.pipelines.vanilla_biometrics.legacy import LegacyBiometricAlgorithm
import os
base_dir = "/idiap/temp/tpereira/mobio/pca"
#base_dir = "./example"
original_directory=rc['bob.db.mobio.directory'] original_directory=rc['bob.db.mobio.directory']
...@@ -54,12 +57,15 @@ from bob.pipelines.mixins import mix_me_up ...@@ -54,12 +57,15 @@ from bob.pipelines.mixins import mix_me_up
preprocessor = mix_me_up((CheckpointMixin, SampleMixin), LegacyProcessorMixin) preprocessor = mix_me_up((CheckpointMixin, SampleMixin), LegacyProcessorMixin)
from bob.pipelines.mixins import dask_it from bob.pipelines.mixins import dask_it
extractor = Pipeline(steps=[('0', preprocessor(callable=original_preprocessor, features_dir="./example/extractor0")), extractor = Pipeline(steps=[
('1',CheckpointSampleLinearize(features_dir="./example/extractor1")), ('0', preprocessor(callable=original_preprocessor, features_dir=os.path.join(base_dir,"extractor0"))),
('2',CheckpointSamplePCA(features_dir="./example/extractor2", model_path="./example/pca.pkl"))]) ('1',CheckpointSampleLinearize(features_dir=os.path.join(base_dir,"extractor1"))),
extractor = dask_it(extractor) ('2',CheckpointSamplePCA(features_dir=os.path.join(base_dir,"extractor2"), model_path=os.path.join(base_dir,"pca.pkl")))
])
#extractor = dask_it(extractor)
from bob.bio.base.pipelines.vanilla_biometrics.biometric_algorithm import Distance, BiometricAlgorithmCheckpointMixin from bob.bio.base.pipelines.vanilla_biometrics.biometric_algorithm import Distance, BiometricAlgorithmCheckpointMixin
class CheckpointDistance(BiometricAlgorithmCheckpointMixin, Distance): pass class CheckpointDistance(BiometricAlgorithmCheckpointMixin, Distance): pass
algorithm = CheckpointDistance(features_dir="./example/") algorithm = CheckpointDistance(features_dir=base_dir)
#algorithm = Distance() #algorithm = Distance()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment