diff --git a/bob/ip/tensorflow_extractor/DrGanMSU.py b/bob/ip/tensorflow_extractor/DrGanMSU.py index 5886a52f8418f95d7919b0d8f49a94b4559bf991..b856ca2598a5f3ccf35c092bf3eb1ee3fed540db 100644 --- a/bob/ip/tensorflow_extractor/DrGanMSU.py +++ b/bob/ip/tensorflow_extractor/DrGanMSU.py @@ -5,7 +5,7 @@ import numpy import tensorflow as tf import os -import pkg_resources +from bob.extension import rc from . import download_file import logging logger = logging.getLogger(__name__) @@ -331,7 +331,7 @@ class DrGanMSUExtractor(object): """ - def __init__(self, model_path=None, image_size=[96, 96, 3]): + def __init__(self, model_path=rc["drgan_modelpath"], image_size=[96, 96, 3]): self.image_size = image_size self.session = tf.Session() diff --git a/bob/ip/tensorflow_extractor/FaceNet.py b/bob/ip/tensorflow_extractor/FaceNet.py index d6e5e1976d6d6fb582b6a4356eacb8303dcbc980..60211362e67a801cea8da04881988e201d0e7960 100644 --- a/bob/ip/tensorflow_extractor/FaceNet.py +++ b/bob/ip/tensorflow_extractor/FaceNet.py @@ -7,6 +7,8 @@ import tensorflow as tf from bob.ip.color import gray_to_rgb from bob.io.image import to_matplotlib from . import download_file +from bob.extension import rc + logger = logging.getLogger(__name__) @@ -72,7 +74,7 @@ class FaceNet(object): """ def __init__(self, - model_path=None, + model_path=rc["facenet_modelpath"], image_size=160, **kwargs): super(FaceNet, self).__init__()