Example: 01-Camera/01-Video-Recording/mjpeg_on_face_detection.py

# 本作品采用MIT许可证授权。
# 版权所有 (c) 2013-2023 OpenMV LLC。保留所有权利。
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# 基于人脸检测的MJPEG视频录制示例
#
# 注意：运行此示例需要一张SD卡。
#
# 你可以使用你的OpenMV Cam录制mjpeg文件。你可以向
# 录制器对象提供JPEG帧或RGB565/灰度帧。一旦你完成
# 录制Mjpeg文件，你可以使用VLC播放它。如果你在Ubuntu上
# 内置的视频播放器也能正常工作。
#
# 此示例演示了如何在OpenMV摄像头上使用面部追踪功能拍摄一张
# MJPEG格式。

import sensor
import image
import time
import mjpeg
import random

sensor.reset()  # 重置并初始化传感器。
sensor.set_pixformat(sensor.GRAYSCALE)  # 将像素格式设置为RGB565 (or GRAYSCALE)
sensor.set_framesize(sensor.QQVGA)  # 将帧大小设置为QQVGA（160x120）
sensor.skip_frames(time=2000)  # 等待设置生效。

led = machine.LED("LED_RED")

# 加载面部检测的HaarCascade。这是您的OpenMV摄像头
# 可以使用下面的find_features()方法来检测人脸。您的OpenMV
# 摄像头内置了frontalface HaarCascade。默认情况下，所有阶段的
# HaarCascade都会被加载。然而，您可以调整阶段数以加快
# 处理速度，但会牺牲准确性。frontalface HaarCascade有25个
# 阶段。
face_cascade = image.HaarCascade("frontalface", stages=25)

while True:
    print("About to start detecting faces...")
    sensor.skip_frames(time=2000)  # 给用户时间准备。

    print("Now detecting faces!")

    diff = 10  # 我们会在10帧后检测到人脸。
    while diff:
        img = sensor.snapshot()
        # 阈值可以在0.0到1.0之间。较高的阈值会导致
        # 更高的检测率，但会有更多的误报。比例值
        # 控制匹配比例，允许您检测更小的面孔。
        faces = img.find_features(face_cascade, threshold=0.5, scale_factor=1.5)

        if faces:
            diff -= 1
            for r in faces:
                img.draw_rectangle(r)

    led.on()
    m = mjpeg.Mjpeg("example-%d.mjpeg" % random.getrandbits(32))

    clock = time.clock()  # 跟踪FPS。
    for i in range(200):
        clock.tick()
        m.write(sensor.snapshot())
        print(clock.fps())

    m.close()
    led.off()
    print("Restarting...")