diff --git a/bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py b/bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py index d5a56c398a8ce56c6ad6cff215fe647920fabdfd..0ed5bf4e9c47b564185e9b0437b43cd8d0d95d43 100644 --- a/bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py +++ b/bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py @@ -74,7 +74,7 @@ class DBGenerator(object): self.cropper = cropper self.extractor = extractor self.seed = seed - with open(rc["bob.synface.latent_directions"], "rb") as f: + with open(covariates_analysis_path, "rb") as f: self.covariates_analysis = pkl.load(f) self.ict = ict diff --git a/bob/paper/ijcb2021_synthetic_dataset/script/generate_db.py b/bob/paper/ijcb2021_synthetic_dataset/script/generate_db.py index 57c2562da9f6a37bac978565f6e74be1126e8e45..9ae0e01ac2869064e120a40dc264ad2041194820 100644 --- a/bob/paper/ijcb2021_synthetic_dataset/script/generate_db.py +++ b/bob/paper/ijcb2021_synthetic_dataset/script/generate_db.py @@ -4,9 +4,7 @@ from bob.bio.face_ongoing.configs.baselines.msceleb.inception_resnet_v2.centerlo from ..generate.db_generator import DBGenerator from ..stylegan2.generator import StyleGAN2Generator from ..stylegan2.dnnlib import tflib -from ..latent import analysis, editing from ..utils import get_task -import numpy as np import pickle as pkl from scipy.spatial.distance import cdist @@ -155,7 +153,7 @@ def initialization( "-o", type=str, required=True, - default=rc['bob.db.synmultipie.directory'], + default=rc['bob.synface.synthetic_datasets'], help="Root of the output directory tree", cls=ResourceOption, ) diff --git a/bob/paper/ijcb2021_synthetic_dataset/utils/__init__.py b/bob/paper/ijcb2021_synthetic_dataset/utils/__init__.py index bad63140a6f1927a9253f58b06b1c08c96c0683a..5088ea49d2b4ccd58bf4470741a2d22be7113f7e 100644 --- a/bob/paper/ijcb2021_synthetic_dataset/utils/__init__.py +++ b/bob/paper/ijcb2021_synthetic_dataset/utils/__init__.py @@ -2,9 +2,9 @@ from pkgutil import extend_path __path__ = extend_path(__path__, __name__) import numpy as np import os -from ..stylegan2.dnnlib import tflib def fix_randomness(seed=None): + from ..stylegan2.dnnlib import tflib config = {'rnd.np_random_seed': seed, 'rnd.tf_random_seed': 'auto'} tflib.init_tf(config)