Commit 8d48fd4a authored by Xinyi ZHANG's avatar Xinyi ZHANG Committed by Tiago de Freitas Pereira
Browse files

”chang_gpu”

parent d6513a81
......@@ -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"),
......
......@@ -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"),
......
......@@ -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"),
......
......@@ -43,8 +43,6 @@ requirements:
- scikit-image {{ scikit_image }}
- six {{ six }}
- tensorflow {{ tensorflow }} # [linux]
- mxnet {{ mxnet }}
- opencv-pyrhon {{ opencv_python }}
run:
- python
- setuptools
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment