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 @@
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()
......
......@@ -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__()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment