Skip to content
Snippets Groups Projects
Commit abee8d3b authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[doc] extended docstrings, removed reference to DRGAN for now

parent 49d8ed0b
No related branches found
No related tags found
No related merge requests found
...@@ -10,7 +10,6 @@ parts ...@@ -10,7 +10,6 @@ parts
src src
develop-eggs develop-eggs
sphinx sphinx
test*
submit* submit*
log* log*
results* results*
......
...@@ -69,17 +69,17 @@ class CNN8Extractor(Extractor): ...@@ -69,17 +69,17 @@ class CNN8Extractor(Extractor):
def __call__(self, image): def __call__(self, image):
"""__call__(image) -> feature """__call__(image) -> feature
Extract features Extract features from an image
**Parameters:** **Parameters:**
image : 3D :py:class:`numpy.ndarray` (floats) image : 3D :py:class:`numpy.ndarray` (floats)
The image to extract the features from. The image to extract the features from. Its size must be 3x128x128
**Returns:** **Returns:**
feature : 2D :py:class:`numpy.ndarray` (floats) feature : 2D :py:class:`numpy.ndarray` (floats)
The extracted features The extracted features as a 1d array of size 512
""" """
input_image = numpy.rollaxis(numpy.rollaxis(image, 2),2) input_image = numpy.rollaxis(numpy.rollaxis(image, 2),2)
input_image = self.to_tensor(input_image) input_image = self.to_tensor(input_image)
......
...@@ -73,12 +73,12 @@ class CasiaNetExtractor(Extractor): ...@@ -73,12 +73,12 @@ class CasiaNetExtractor(Extractor):
**Parameters:** **Parameters:**
image : 3D :py:class:`numpy.ndarray` (floats) image : 3D :py:class:`numpy.ndarray` (floats)
The image to extract the features from. The image to extract the features from. Its size must be 3x128x128
**Returns:** **Returns:**
feature : 2D :py:class:`numpy.ndarray` (floats) feature : 2D :py:class:`numpy.ndarray` (floats)
The extracted features The extracted features as a 1d array of size 320
""" """
input_image = numpy.rollaxis(numpy.rollaxis(image, 2),2) input_image = numpy.rollaxis(numpy.rollaxis(image, 2),2)
input_image = self.to_tensor(input_image) input_image = self.to_tensor(input_image)
......
=========== ==========
User guide User guide
=========== ==========
Using as a feature extractor Using as a feature extractor
---------------------------- ----------------------------
...@@ -9,7 +9,7 @@ This package has been done with the intent of using it in conjunction with ...@@ -9,7 +9,7 @@ This package has been done with the intent of using it in conjunction with
bob.bio.face to run face verification experiments. bob.bio.face to run face verification experiments.
In particular, the goal here is to encode the identity contained in a In particular, the goal here is to encode the identity contained in a
face image using a model referred to as DR-GAN face image using a different CNN models
For this purpose, you can specify your feature extractor in configuration For this purpose, you can specify your feature extractor in configuration
file to be used together with the ```verifiy.py``` script from ```bob.bio.base``` file to be used together with the ``verifiy.py`` script from ``bob.bio.base``
...@@ -6,7 +6,7 @@ ...@@ -6,7 +6,7 @@
=================================================== ===================================================
This package contains functionality to extract features from models trained with This package contains functionality to extract features from models trained with
Tensorflow http://pytorch.org/ PyTorch http://pytorch.org/
======= =======
Index Index
......
...@@ -7,8 +7,8 @@ Classes ...@@ -7,8 +7,8 @@ Classes
------- -------
.. autosummary:: .. autosummary::
bob.ip.pytorch_extractor.DRGANLight bob.ip.pytorch_extractor.CNN8
bob.ip.pytorch_extractor.DRGANOriginal bob.ip.pytorch_extractor.CasiaNet
Detailed API Detailed API
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment