RTSP视频服务器

# 本作品采用MIT许可证授权。
# 版权所有 (c) 2013-2023 OpenMV LLC。保留所有权利。
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# RTSP视频服务器
#
# 此示例展示了如何用OpenMV摄像头通过RTSP推流视频。
#
# 您可以使用VLC之类的程序连接到OpenMV摄像头的IP地址来观看视频流。
# 不过，OpenMV IDE内置了基于FFPLAY的RTSP查看器，
# 您可以通过菜单 工具->视频工具->播放RTSP流 来使用它。

import network
import rtsp
import csi
import time

# 如果您在Linux上使用VLC，可能需要安装live555库以支持RTSP功能。
# 若您使用的是Ubuntu系统，请运行以下命令：
#
# sudo apt-get install livemedia-utils

# 关于VLC等程序的延迟问题，默认通常会缓冲1秒的视频
# 才开始播放。要减少延迟，您需要降低网络缓存设置，这可以通过
# 在VLC中打开网络流时选择“显示更多选项”来调整。您可以将缓存时间降至10毫秒
# 以实现视频的实时播放。

csi0 = csi.CSI(stream=False)
csi0.reset()
csi0.pixformat(csi.RGB565)
csi0.framesize(csi.VGA)

# 设置网络接口

network_if = network.WLAN(network.STA_IF)
network_if.active(True)
network_if.connect("your-ssid", "your-password")
while not network_if.isconnected():
    print("Trying to connect. Note this may take a while...")
    time.sleep_ms(1000)

# 设置RTSP服务器

server = rtsp.rtsp_server(network_if)

# 对于以下回调函数：
#
# `pathname`是客户端请求的流资源名称。如果不需要，您可以忽略此参数。
# 否则，您可以用它来决定返回哪个图像对象。默认情况下，路径
# 名称会是“/”。
#
# `session`是一个随机数，当新连接建立时会改变。您可以利用
# session配合字典来区分对同一文件名的不同访问。

# 追踪当前帧率。
clock = time.clock()


def setup_callback(pathname, session):
    print('Opening "%s" in session %d' % (pathname, session))


def play_callback(pathname, session):
    global clock
    clock.reset()
    clock.tick()
    print('Playing "%s" in session %d' % (pathname, session))


def pause_callback(pathname, session):  # VLC 仅在本地暂停。此功能从未被调用。
    print('Pausing "%s" in session %d' % (pathname, session))


def teardown_callback(pathname, session):
    print('Closing "%s" in session %d' % (pathname, session))


server.register_setup_cb(setup_callback)
server.register_play_cb(play_callback)
server.register_pause_cb(pause_callback)
server.register_teardown_cb(teardown_callback)


# 每次需要新帧时调用。
def image_callback(pathname, session):
    global csi0
    global clock
    img = csi0.snapshot()
    # 标记图像和/或执行各种操作。
    print(clock.fps())
    clock.tick()
    return img


# 流不会返回。当需要获取图像对象发送时，它将调用 `image_callback`
# 给连接到服务器的远程 RTSP 客户端。

server.stream(image_callback, quality=70)