diff --git a/bob/bio/face/extractor/opencv_caffe.py b/bob/bio/face/extractor/opencv_caffe.py index 7634e4c13a870d297119515ae219678372759916..53ab52a6511985553ce6c70211995dac025a92fa 100644 --- a/bob/bio/face/extractor/opencv_caffe.py +++ b/bob/bio/face/extractor/opencv_caffe.py @@ -40,14 +40,12 @@ class opencv_model(TransformerMixin, BaseEstimator): .. note:: This structure only can be used for CAFFE pretrained model. - **Parameters:** - use_gpu: True or False. + """ - def __init__(self, use_gpu=False, **kwargs): + def __init__(self, **kwargs): super().__init__(**kwargs) self.model = None - self.use_gpu = use_gpu internal_path = pkg_resources.resource_filename( __name__, os.path.join("data", "opencv_model"), diff --git a/bob/bio/face/extractor/pytorch_model.py b/bob/bio/face/extractor/pytorch_model.py index d784d7f48e6cd9c6eae1607b2a103940a01edfb7..2d7a5f1c14e263aac1c1eeed237afb465cee89ee 100644 --- a/bob/bio/face/extractor/pytorch_model.py +++ b/bob/bio/face/extractor/pytorch_model.py @@ -29,14 +29,12 @@ class pytorch_loaded_model(TransformerMixin, BaseEstimator): The extracted features can be combined with different the algorithms. - **Parameters:** - use_gpu: True or False. + """ - def __init__(self, use_gpu=False, **kwargs): + def __init__(self, **kwargs): super().__init__(**kwargs) self.model = None - self.use_gpu = use_gpu internal_path = pkg_resources.resource_filename( __name__, os.path.join("data", "resnet"), @@ -110,10 +108,9 @@ class pytorch_library_model(TransformerMixin, BaseEstimator): use_gpu: True or False. """ - def __init__(self, model=None, use_gpu=False, **kwargs): + def __init__(self, model=None, **kwargs): super().__init__(**kwargs) self.model = model - self.use_gpu = use_gpu internal_path = pkg_resources.resource_filename( __name__, os.path.join("data", "resnet"), diff --git a/bob/bio/face/extractor/tf_model.py b/bob/bio/face/extractor/tf_model.py index ae39ebb45f65cd8601715f2de88e890c727df245..c690f2acb339c903aa56367e7ceffac8996062c9 100644 --- a/bob/bio/face/extractor/tf_model.py +++ b/bob/bio/face/extractor/tf_model.py @@ -30,14 +30,11 @@ class tf_model(TransformerMixin, BaseEstimator): The extracted features can be combined with different the algorithms. - **Parameters:** - use_gpu: True or False. """ - def __init__(self, use_gpu=False, **kwargs): + def __init__(self, **kwargs): super().__init__(**kwargs) self.model = None - self.use_gpu = use_gpu internal_path = pkg_resources.resource_filename( __name__, os.path.join("data", "resnet"), diff --git a/conda/meta.yaml b/conda/meta.yaml index 129c3dc245bd4919797d498c55bcbb7753cb9c88..6250242cefc4bbd94d530964896e8a0b02cc7957 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -43,8 +43,6 @@ requirements: - scikit-image {{ scikit_image }} - six {{ six }} - tensorflow {{ tensorflow }} # [linux] - - mxnet {{ mxnet }} - - opencv-pyrhon {{ opencv_python }} run: - python - setuptools