enroll_image.py

"""
License: Apache 2.0. See LICENSE file in root directory.
Copyright(c) 2020-2021 Intel Corporation. All Rights Reserved.
"""
import sys
import math
import cv2
import rsid_py

PORT="COM9"

def resize_to_120(cv_image):
    height, width, channels = cv_image.shape
    if width <= 120 and height <= 120:
        return cv_image
    if width > height:
        new_width = 120
        scale_h = new_width / width
        new_height = int(height * scale_h)
    else:
        new_height = 120
        scale_w = new_height / height
        new_width = int(width * scale_w)

    print(f"Resize to {new_width}x{new_height}")
    return cv2.resize(cv_image, (new_width, new_height), interpolation=cv2.INTER_CUBIC)
    
    
def enroll_with_image(user_id, filename):
    im_cv = cv2.imread(filename)
    im_cv = resize_to_120(im_cv)
    height, width, channels = im_cv.shape
    with rsid_py.FaceAuthenticator(PORT) as f:
        result = f.enroll_image(user_id, im_cv.flatten(), width, height)
        print(result)


if __name__ == '__main__':
    if len(sys.argv) != 3:
        print("Usage: python enroll_image.py <user-id> <image-filename>")
        sys.exit(1)
    enroll_with_image(sys.argv[1], sys.argv[2])