align_dlib: Add returns.

2016-06-10 11:05:20 -04:00 · 2016-06-10 11:05:20 -04:00 · 5e0ab14049
parent bb2ef270c1
commit 5e0ab14049
1 changed files with 2 additions and 2 deletions
--- a/openface/align_dlib.py
+++ b/openface/align_dlib.py
@ -173,9 +173,9 @@ class AlignDlib:
        :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