Example: nodes/models/gaussian.py

# Copyright (c) 2024 Justin Davis (davisjustin302@gmail.com)
#
# MIT License
"""Example showcasing how to use the Gaussian model."""

from __future__ import annotations

import cv2
import depthai as dai
from oakutils.nodes import create_color_camera, create_xout, get_nn_bgr_frame
from oakutils.nodes.models import create_gaussian

pipeline = dai.Pipeline()

# create the color camera node
cam = create_color_camera(pipeline, preview_size=(640, 480))

# create neural network node
lp = create_gaussian(pipeline, cam.preview, kernel_size=5)
xout_lp = create_xout(pipeline, lp.out, "gaussian")

with dai.Device(pipeline) as device:
    l_queue: dai.DataOutputQueue = device.getOutputQueue("gaussian")

    while True:
        l_data = l_queue.get()
        l_frame = get_nn_bgr_frame(l_data, frame_size=(640, 480), normalization=255.0)

        cv2.imshow("gaussian frame", l_frame)
        if cv2.waitKey(1) == ord("q"):
            break