”chang_gpu”

bob/bio/face/extractor/opencv_caffe.py

@@ -40,15 +40,14 @@ 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"),

bob/bio/face/extractor/pytorch_model.py

@@ -28,14 +28,13 @@ 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"),
@@ -109,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"),

bob/bio/face/extractor/tf_model.py

@@ -29,14 +29,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"),

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