Skip to content
Snippets Groups Projects
Commit ade99880 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Merge branch 'dev_branch' into 'master'

Fix for the "empty image" issue

See merge request !3
parents 71c5b8d0 95d580d3
No related branches found
No related tags found
1 merge request!3Fix for the "empty image" issue
Pipeline #
......@@ -58,7 +58,7 @@ class FaceDetector(object):
def detect_single_face(self, image):
faces = self.detect_all_faces(image)
if len(faces) > 1:
if len(faces) > 1 and not all([not f for f in faces]):
index = numpy.argmax([(f.bottomright[0] - f.topleft[0]) * (f.bottomright[1] - f.topleft[1]) for f in faces[0]])
return (faces[0][index], faces[1][index])
else:
......
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
"""Test Units
"""
#==============================================================================
# Import what is needed here:
import numpy as np
from .. import FaceDetector
def test_face_detector():
"""
Test FaceDetector class.
"""
image = np.zeros((3, 100, 100))
result = FaceDetector().detect_single_face(image)
assert result is None
image = np.ones((3, 100, 100))
result = FaceDetector().detect_single_face(image)
assert result is None
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment