Commit 1704ccec authored by Richard Torenvliet's avatar Richard Torenvliet

Update some documentation

parent 20492cc6
...@@ -31,11 +31,23 @@ def test_detect(image): ...@@ -31,11 +31,23 @@ def test_detect(image):
class Detector(): class Detector():
"""
Use dlib library to detect landmarks in a given image
"""
def __init__(self): def __init__(self):
self.detector = dlib.get_frontal_face_detector() self.detector = dlib.get_frontal_face_detector()
self.predictor = dlib.shape_predictor(LANDMARK_DETECTOR_PATH) self.predictor = dlib.shape_predictor(LANDMARK_DETECTOR_PATH)
def detect_faces(self, image): def detect_faces(self, image):
"""
Detect faces in an image.
Args:
image: np.array
Returns:
array of detection arrays.
"""
# The 1 in the second argument indicates that we should upsample the # The 1 in the second argument indicates that we should upsample the
# image 1 time. This will make everything bigger and allow us to # image 1 time. This will make everything bigger and allow us to
# detect more faces. # detect more faces.
......
...@@ -17,7 +17,7 @@ from utility import import_dataset_module ...@@ -17,7 +17,7 @@ from utility import import_dataset_module
def save_pca_model_texture(args): def save_pca_model_texture(args):
""" """
save the U, s, Vt and mean of all the asf datafiles given by the asf Save the U, s, Vt and mean of all the asf datafiles given by the asf
files. files.
It is saved in the following way: It is saved in the following way:
......
Markdown is supported
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