Skip to content
Snippets Groups Projects
Commit 3f4f126c authored by Laurent COLBOIS's avatar Laurent COLBOIS
Browse files

Add config entry point for database projection

parent 3fa95e4d
Branches
No related tags found
No related merge requests found
...@@ -28,15 +28,16 @@ def is_processed(f, output_dir): ...@@ -28,15 +28,16 @@ def is_processed(f, output_dir):
@click.command( @click.command(
cls=ConfigCommand, cls=ConfigCommand,
entry_point_group='projection_config',
help="Project a bob.bio.base.database.BioDatabase into the StyleGAN2 latent space.", help="Project a bob.bio.base.database.BioDatabase into the StyleGAN2 latent space.",
) )
@click.option( @click.option(
"--database", "--database",
"-d", "-d",
entry_point_group="projected_db", entry_point_group='projected_db',
required=True, required=True,
cls=ResourceOption, cls=ResourceOption,
help="Which bob.bio.database to project (available : `multipie_U`, `multipie_E`, `multipie_P`)", help="Which bob.bio.database to project (available : 'multipie_U', 'multipie_E', 'multipie_P')"
) )
@click.option( @click.option(
"--output-dir", "--output-dir",
......
...@@ -45,6 +45,11 @@ setup( ...@@ -45,6 +45,11 @@ setup(
'latent_analysis.py = bob.paper.ijcb2021_synthetic_dataset.script.compute_latent_directions:compute_latents', 'latent_analysis.py = bob.paper.ijcb2021_synthetic_dataset.script.compute_latent_directions:compute_latents',
'generate_db.py = bob.paper.ijcb2021_synthetic_dataset.script.generate_db:db_gen' 'generate_db.py = bob.paper.ijcb2021_synthetic_dataset.script.generate_db:db_gen'
], ],
"projection_config": [
'multipie_U = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_U',
'multipie_E = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_E',
'multipie_P = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_P',
],
"projected_db": [ "projected_db": [
'multipie_U = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_U:database', 'multipie_U = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_U:database',
'multipie_E = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_E:database', 'multipie_E = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_E:database',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment