From 70dba047ab95a02d4eb05d92cf4df752996db2b2 Mon Sep 17 00:00:00 2001 From: Laurent COLBOIS <laurent.colbois@idiap.ch> Date: Fri, 18 Jun 2021 09:57:40 +0200 Subject: [PATCH] Path tweaks --- bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py | 2 +- bob/paper/ijcb2021_synthetic_dataset/script/generate_db.py | 4 +--- bob/paper/ijcb2021_synthetic_dataset/utils/__init__.py | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py b/bob/paper/ijcb2021_synthetic_dataset/generate/db_generator.py index d5a56c3..0ed5bf4 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 57c2562..9ae0e01 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 bad6314..5088ea4 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) -- GitLab