diff --git a/bob/paper/ijcb2021_synthetic_dataset/script/project_db.py b/bob/paper/ijcb2021_synthetic_dataset/script/project_db.py index 3977161d7232f1b7f1c28c762f2ab0a0446abd97..fa5cb42aec626406048a289f5a65d72296504160 100644 --- a/bob/paper/ijcb2021_synthetic_dataset/script/project_db.py +++ b/bob/paper/ijcb2021_synthetic_dataset/script/project_db.py @@ -28,15 +28,16 @@ def is_processed(f, output_dir): @click.command( cls=ConfigCommand, + entry_point_group='projection_config', help="Project a bob.bio.base.database.BioDatabase into the StyleGAN2 latent space.", ) @click.option( "--database", "-d", - entry_point_group="projected_db", + entry_point_group='projected_db', required=True, 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( "--output-dir", diff --git a/setup.py b/setup.py index 466641ba219a77dec8467a08936db149c1978e12..8fd8ec0ed371e8fbc0a0f751ab91e669e9f1f26c 100644 --- a/setup.py +++ b/setup.py @@ -45,6 +45,11 @@ setup( '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' ], + "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": [ 'multipie_U = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_U:database', 'multipie_E = bob.paper.ijcb2021_synthetic_dataset.config.project.multipie_E:database',