From 009e21a8a640467c3bd0069ec75975cc0ba6219e Mon Sep 17 00:00:00 2001 From: Amir MOHAMMADI <amir.mohammadi@idiap.ch> Date: Mon, 12 Mar 2018 14:59:12 +0100 Subject: [PATCH] Test min_face_size_validator --- bob/bio/face/annotator/__init__.py | 4 +++- bob/bio/face/test/test_annotators.py | 20 +++++++++++++++++++- 2 files changed, 22 insertions(+), 2 deletions(-) diff --git a/bob/bio/face/annotator/__init__.py b/bob/bio/face/annotator/__init__.py index ce553638..e4969dac 100644 --- a/bob/bio/face/annotator/__init__.py +++ b/bob/bio/face/annotator/__init__.py @@ -36,7 +36,9 @@ def min_face_size_validator(annotations, min_face_size=(32, 32)): bool True, if the face is large enough. """ - bbx = bob.ip.facedetect.bounding_box_from_annotations( + if not annotations: + return False + bbx = bob.ip.facedetect.bounding_box_from_annotation( source='direct', **annotations) if bbx.size < min_face_size: return False diff --git a/bob/bio/face/test/test_annotators.py b/bob/bio/face/test/test_annotators.py index 4f71886e..e35c0a4a 100644 --- a/bob/bio/face/test/test_annotators.py +++ b/bob/bio/face/test/test_annotators.py @@ -2,7 +2,8 @@ import bob.io.base import bob.io.base.test_utils import bob.io.image from bob.bio.face.annotator import ( - BobIpFacedetect, BoundingBoxToEyes, BobIpFlandmark) + BobIpFacedetect, BoundingBoxToEyes, BobIpFlandmark, + min_face_size_validator) from bob.bio.base.annotator import FailSafe import numpy @@ -46,3 +47,20 @@ def test_bob_ip_flandmark(): _assert_bob_ip_facedetect(annot) assert [int(x) for x in annot['reye']] == [183, 127], annot assert [int(x) for x in annot['leye']] == [174, 223], annot + + +def test_min_face_size_validator(): + valid = { + 'topleft': (0, 0), + 'bottomright': (32, 32), + } + assert min_face_size_validator(valid) + + not_valid = { + 'topleft': (0, 0), + 'bottomright': (28, 32), + } + assert not min_face_size_validator(not_valid) + + assert not min_face_size_validator(None) + assert not min_face_size_validator({}) -- GitLab