thermal_overlay.py

# Dieses Werk ist unter der MIT-Lizenz lizenziert.
# Copyright (c) 2013-2023 OpenMV LLC. Alle Rechte vorbehalten.
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# Demo für Wärmebild-Overlay
#
# Dieses Beispiel zeigt, wie eine Heatmap auf die
# Live-Videoausgabe der Hauptkamera Ihrer OpenMV Cam überlagert wird.

import csi
import image
import time
import fir

# oder image.BILINEAR oder 0 (nächster Nachbar)
drawing_hint = image.BICUBIC | image.CENTER | image.SCALE_ASPECT_KEEP

ALT_OVERLAY = False  # Auf True setzen, um ein zweites IR-Bild zu allokieren.

csi0 = csi.CSI()
csi0.reset()
csi0.pixformat(csi.RGB565)
csi0.framesize(csi.QQVGA)
csi0.snapshot(time=2000)

# Wärmebildsensor initialisieren
fir.init()

# Einen weiteren Framebuffer für ein flüssigeres Video allokieren.
extra_fb = image.Image(csi0.width(), csi0.height(), csi0.pixformat())

# FPS-Uhr
clock = time.clock()

while True:
    clock.tick()

    # Ein Bild aufnehmen
    img = csi0.snapshot()

    # FIR-Daten erfassen
    # ta: Umgebungstemperatur
    # ir: Objekttemperaturen (IR-Array)
    # to_min: Minimale Objekttemperatur
    # to_max: Maximale Objekttemperatur
    try:
        ta, ir, to_min, to_max = fir.read_ir()
    except OSError:
        continue

    if not ALT_OVERLAY:
        # Bild skalieren und mit dem Framebuffer überblenden
        fir.draw_ir(img, ir, hint=drawing_hint)
    else:
        # Ein zweites Bild erzeugen und dann in den Framebuffer einblenden.
        extra_fb.clear()
        fir.draw_ir(extra_fb, ir, hint=drawing_hint)
        img.blend(extra_fb, alpha=128)

    # Umgebungs-, Minimal- und Maximaltemperatur einzeichnen.
    img.draw_string((8, 0), "Ta: %0.2f C" % ta, color=(255, 0, 0), mono_space=False)
    img.draw_string(
        (8, 8), "To min: %0.2f C" % to_min, color=(255, 0, 0), mono_space=False
    )
    img.draw_string(
        (8, 16), "To max: %0.2f C" % to_max, color=(255, 0, 0), mono_space=False
    )

    # Hochwertiges Streaming erzwingen ...
    img.to_jpeg(quality=90)

    # FPS ausgeben.
    print(clock.fps())