[LightCNN9] improved docstrings

bob/ip/pytorch_extractor/LightCNN9.py

@@ -65,19 +65,18 @@ class LightCNN9Extractor(Extractor):
 
     Parameters
     ----------
-    image : 3D :py:class:`numpy.ndarray` (floats)
-      The image to extract the features from. Its size must be 3x128x128
+    image : 2D :py:class:`numpy.ndarray` (floats)
+      The grayscale image to extract the features from. Its size must be 128x128
 
     Returns
     -------
-    feature : 2D :py:class:`numpy.ndarray` (floats)
+    feature : :py:class:`numpy.ndarray` (floats)
       The extracted features as a 1d array of size 320 
     
     """
   
     # torchvision.transforms expect a numpy array of size HxWxC
     input_image = numpy.expand_dims(image, axis=2)
-    
     input_image = self.to_tensor(input_image)
     input_image = self.norm(input_image)
     input_image = input_image.unsqueeze(0)