1 # Copyright 2014 The Android Open Source Project 2 # 3 # Licensed under the Apache License, Version 2.0 (the "License"); 4 # you may not use this file except in compliance with the License. 5 # You may obtain a copy of the License at 6 # 7 # http://www.apache.org/licenses/LICENSE-2.0 8 # 9 # Unless required by applicable law or agreed to in writing, software 10 # distributed under the License is distributed on an "AS IS" BASIS, 11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 # See the License for the specific language governing permissions and 13 # limitations under the License. 14 15 import its.image 16 import its.device 17 import its.objects 18 import os.path 19 20 def main(): 21 """Test face detection. 22 """ 23 NAME = os.path.basename(__file__).split(".")[0] 24 NUM_TEST_FRAMES = 20 25 FD_MODE_OFF = 0 26 FD_MODE_SIMPLE = 1 27 FD_MODE_FULL = 2 28 29 with its.device.ItsSession() as cam: 30 props = cam.get_camera_properties() 31 fd_modes = props['android.statistics.info.availableFaceDetectModes'] 32 a = props['android.sensor.info.activeArraySize'] 33 aw, ah = a['right'] - a['left'], a['bottom'] - a['top'] 34 cam.do_3a() 35 for fd_mode in fd_modes: 36 assert(FD_MODE_OFF <= fd_mode <= FD_MODE_FULL) 37 req = its.objects.auto_capture_request() 38 req['android.statistics.faceDetectMode'] = fd_mode 39 caps = cam.do_capture([req]*NUM_TEST_FRAMES) 40 for i,cap in enumerate(caps): 41 md = cap['metadata'] 42 assert(md['android.statistics.faceDetectMode'] == fd_mode) 43 faces = md['android.statistics.faces'] 44 45 # 0 faces should be returned for OFF mode 46 if fd_mode == FD_MODE_OFF: 47 assert(len(faces) == 0) 48 continue 49 # Face detection could take several frames to warm up, 50 # but it should detect at least one face in last frame 51 if i == NUM_TEST_FRAMES - 1: 52 if len(faces) == 0: 53 print "Error: no face detected in mode", fd_mode 54 assert(0) 55 if len(faces) == 0: 56 continue 57 58 print "Frame %d face metadata:" % i 59 print " Faces:", faces 60 print "" 61 62 face_scores = [face['score'] for face in faces] 63 face_rectangles = [face['bounds'] for face in faces] 64 for score in face_scores: 65 assert(score >= 1 and score <= 100) 66 # Face bounds should be within active array 67 for rect in face_rectangles: 68 assert(rect['top'] < rect['bottom']) 69 assert(rect['left'] < rect['right']) 70 assert(0 <= rect['top'] <= ah) 71 assert(0 <= rect['bottom'] <= ah) 72 assert(0 <= rect['left'] <= aw) 73 assert(0 <= rect['right'] <= aw) 74 75 # Face landmarks are reported if and only if fd_mode is FULL 76 # Face ID should be -1 for SIMPLE and unique for FULL 77 if fd_mode == FD_MODE_SIMPLE: 78 for face in faces: 79 assert('leftEye' not in face) 80 assert('rightEye' not in face) 81 assert('mouth' not in face) 82 assert(face['id'] == -1) 83 elif fd_mode == FD_MODE_FULL: 84 face_ids = [face['id'] for face in faces] 85 assert(len(face_ids) == len(set(face_ids))) 86 # Face landmarks should be within face bounds 87 for face in faces: 88 left_eye = face['leftEye'] 89 right_eye = face['rightEye'] 90 mouth = face['mouth'] 91 l, r = face['bounds']['left'], face['bounds']['right'] 92 t, b = face['bounds']['top'], face['bounds']['bottom'] 93 assert(l <= left_eye['x'] <= r) 94 assert(t <= left_eye['y'] <= b) 95 assert(l <= right_eye['x'] <= r) 96 assert(t <= right_eye['y'] <= b) 97 assert(l <= mouth['x'] <= r) 98 assert(t <= mouth['y'] <= b) 99 100 if __name__ == '__main__': 101 main() 102 103
