Skip to content
Snippets Groups Projects
Commit 81aae6a2 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Merge branch 'global-config' into 'master'

Added support to the new bob.extension global config. Added the variables…

See merge request !5
parents 5add1ae9 ebb01313
No related branches found
No related tags found
1 merge request!5Added support to the new bob.extension global config. Added the variables…
Pipeline #
...@@ -5,7 +5,7 @@ ...@@ -5,7 +5,7 @@
import numpy import numpy
import tensorflow as tf import tensorflow as tf
import os import os
import pkg_resources from bob.extension import rc
from . import download_file from . import download_file
import logging import logging
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -331,7 +331,7 @@ class DrGanMSUExtractor(object): ...@@ -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.image_size = image_size
self.session = tf.Session() self.session = tf.Session()
......
...@@ -7,6 +7,8 @@ import tensorflow as tf ...@@ -7,6 +7,8 @@ import tensorflow as tf
from bob.ip.color import gray_to_rgb from bob.ip.color import gray_to_rgb
from bob.io.image import to_matplotlib from bob.io.image import to_matplotlib
from . import download_file from . import download_file
from bob.extension import rc
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -72,7 +74,7 @@ class FaceNet(object): ...@@ -72,7 +74,7 @@ class FaceNet(object):
""" """
def __init__(self, def __init__(self,
model_path=None, model_path=rc["facenet_modelpath"],
image_size=160, image_size=160,
**kwargs): **kwargs):
super(FaceNet, self).__init__() super(FaceNet, self).__init__()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment