Comparison app. WIP

bob/bio/base/script/compare_samples.py

@@ -19,7 +19,8 @@ import logging
 import os
 import itertools
 import dask.bag
-#from bob.bio.base.pipelines.vanilla_biometrics import (
+
+# from bob.bio.base.pipelines.vanilla_biometrics import (
 #    VanillaBiometricsPipeline,
 #    BioAlgorithmCheckpointWrapper,
 #    BioAlgorithmDaskWrapper,
@@ -28,13 +29,14 @@ import dask.bag
 #    dask_get_partition_size,
 #    FourColumnsScoreWriter,
 #    CSVScoreWriter,
-#)
-#from dask.delayed import Delayed
-#import pkg_resources
+# )
+# from dask.delayed import Delayed
+# import pkg_resources
 from bob.extension.config import load as chain_load
 from bob.pipelines.utils import isinstance_nested
 from bob.bio.base.utils import get_resource_filename
 from .vanilla_biometrics import compute_scores, load_database_pipeline
+from bob.pipelines import Sample, SampleSet
 
 
 logger = logging.getLogger(__name__)
@@ -50,16 +52,15 @@ EPILOG = """\b
 """
 
 
-@click.command(
-    entry_point_group="bob.bio.pipeline", cls=ConfigCommand, epilog=EPILOG,
-)
+@click.command(epilog=EPILOG)
+@click.argument("samples", nargs=-1)
 @click.option(
     "--pipeline",
     "-p",
     required=True,
     cls=ResourceOption,
     entry_point_group="bob.bio.pipeline",
-    help="Vanilla biometrics pipeline composed of a scikit-learn Pipeline and a BioAlgorithm"
+    help="Vanilla biometrics pipeline composed of a scikit-learn Pipeline and a BioAlgorithm",
 )
 @click.option(
     "--dask-client",
@@ -68,32 +69,32 @@ EPILOG = """\b
     cls=ResourceOption,
     help="Dask client for the execution of the pipeline.",
 )
-@click.option(
-    "--samples", "-s", required=True, help="Vanilla biometrics pipeline", multiple=True,
-)
 @verbosity_option(cls=ResourceOption)
 def compare_samples(
-    pipeline,
-    dask_client,
-    samples,
-    **kwargs,
+    samples, pipeline, dask_client, **kwargs,
 ):
     """Compare several samples all vs all using one vanilla biometrics pipeline
 
     """
 
     if len(samples) == 1:
-        raise ValueError("It's necessary to have at least two samples for the comparison")
-
-    #for e in samples:
-        #A = bob.io.base.load(e)
-        #for p in samples:
-            #B = bob.io.base.load(p)
-
-            
-
-
-
+        raise ValueError(
+            "It's necessary to have at least two samples for the comparison"
+        )
+
+    sample_sets = [
+        SampleSet([Sample(bob.io.base.load(s), key=str(i))], key=str(i))
+        for i, s in enumerate(samples)
+    ]
+
+    import ipdb; ipdb.set_trace()
+    for e in sample_sets:
+        biometric_references = pipeline.create_biometric_reference([e])
+        scores = pipeline.compute_scores(biometric_references, sample_sets)
+        pass
+
+    #    B = bob.io.base.load(p)
+    #    pipeline.biometric_algorithm
 
     if dask_client is not None:
         dask_client.shutdown()