首先,您需要在 detect.py 中导入 cv2 库,用来读取和处理 RTSP 流。然后,您可以使用 cv2.VideoCapture() 函数打开 RTSP 流,并使用 while 循环不断读取帧。在每次循环中,您可以使用 YOLO 模型检测图像中的物体,并使用 cv2.imwrite() 函数将当前帧保存到硬盘上。如果检测到目标物体,您可以使用 cv2.imshow() 函数显示当前帧。还可以使用 cv2.VideoWriter() 函数将指定帧数的视频保存到硬盘上。最后,您可以在函数中返回帧的图像和视频。
示例代码如下:
import cv2
def detect_on_