preview_image.py

import cv2
import numpy as np
import resvg_python

from tests.utils import overlay_webcam
from src.svg.model import Model
from src.tracking.tracking import Tracking

if __name__ == "__main__":
    with open("tests/models/face.svg", "r") as model_file:
        model = Model(model_file.read())

    tracking = Tracking()
    tracking.set_model(model)

    frame = cv2.imread("tests/images/test_normal.jpeg")
    # frame = cv2.imread("tests/images/test_closed_eyes.jpeg")
    # frame = cv2.imread("tests/images/test_irises.jpeg")
    # frame = cv2.imread("tests/images/test_eyes_mouth.jpeg")
    # frame = cv2.imread("tests/images/test_head_tilted_2.jpeg")
    # frame = cv2.imread("tests/images/test_lowlight.jpeg")
    # frame = cv2.imread("tests/images/test_lowlight_2.jpeg")

    tracking.process(frame)

    while tracking.last_frame_ms is None:
        pass

    image_bytes = resvg_python.svg_to_png(model.tostring())
    image_decoded = cv2.imdecode(np.array(bytearray(image_bytes)), cv2.IMREAD_COLOR)

    final_image = overlay_webcam(frame, image_decoded)

    while True:
        cv2.imshow("rein", final_image)

        if cv2.waitKey(1) & 0xFF == ord("q"):
            cv2.destroyAllWindows()
            break