align_dlib: Add returns.

cmusatyalab · Jun 10, 2016 · 5e0ab14 · 5e0ab14
1 parent bb2ef27
commit 5e0ab14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openface/align_dlib.py b/openface/align_dlib.py
@@ -173,9 +173,9 @@ def align(self, imgDim, rgbImg, bb=None,
         :rtype: numpy.ndarray
         """
         if version == 1:
-            self.align_v1(imgDim, rgbImg, bb, landmarks, landmarkIndices, skipMulti)
+            return self.align_v1(imgDim, rgbImg, bb, landmarks, landmarkIndices, skipMulti)
         elif version == 2:
-            self.align_v2(imgDim, rgbImg, bb, landmarks, landmarkIndices, skipMulti)
+            return self.align_v2(imgDim, rgbImg, bb, landmarks, landmarkIndices, skipMulti)
         else:
             assert False