diff options
| -rw-r--r-- | megapixels/app/processors/face_recognition.py | 5 | ||||
| -rw-r--r-- | megapixels/commands/demo/all.py (renamed from megapixels/commands/demo/3d_landmark_anim.py) | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/megapixels/app/processors/face_recognition.py b/megapixels/app/processors/face_recognition.py index e0b9f752..a56b1634 100644 --- a/megapixels/app/processors/face_recognition.py +++ b/megapixels/app/processors/face_recognition.py @@ -33,7 +33,10 @@ class RecognitionDLIB: def vec(self, im, bbox, width=100, jitters=cfg.DLIB_FACEREC_JITTERS, padding=cfg.DLIB_FACEREC_PADDING): - # Converts image and bbox into 128d vector + '''Converts image and bbox into 128d vector + :param im: (numpy.ndarray) BGR image + :param bbox: (BBox) + ''' # scale the image so the face is always 100x100 pixels #self.log.debug('compute scale') diff --git a/megapixels/commands/demo/3d_landmark_anim.py b/megapixels/commands/demo/all.py index 22e09297..e447492b 100644 --- a/megapixels/commands/demo/3d_landmark_anim.py +++ b/megapixels/commands/demo/all.py @@ -184,7 +184,7 @@ def cli(ctx, opt_fp_in, opt_fp_out, opt_gpu, opt_gif_frames, draw_utils.draw_degrees(im_pose, pose_data) # draw animated GIF - im = Image.open(opt_fp_out) + im = Image.open(fp_out) im_frames = [] duration = im.info['duration'] try: |