RTSP and RTP – The Backbone of Video Streaming

import cv2
import time

def read_rtsp_stream(rtsp_url: str, max_frames: int = 30):
    """RTSP 스트림에서 프레임을 읽어 저장합니다."""

    # VideoCapture가 RTSP URL을 받아 스트림 연결
    # 내부적으로 OpenCV가 RTSP/RTP 핸드셰이크를 처리함
    cap = cv2.VideoCapture(rtsp_url)

    if not cap.isOpened():
        print(f"스트림 연결 실패: {rtsp_url}")
        return

    print(f"스트림 연결 성공!")
    print(f"  해상도: {int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))}x"
          f"{int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))}")
    print(f"  FPS: {cap.get(cv2.CAP_PROP_FPS)}")

    frames_read = 0
    start_time = time.time()

    while frames_read < max_frames:
        ret, frame = cap.read()  # ret=False면 스트림 끊김

        if not ret:
            print("스트림 끊김 — 재연결 시도...")
            cap.release()
            cap = cv2.VideoCapture(rtsp_url)   # 재연결
            continue

        frames_read += 1
        # 실제 AI 파이프라인에서는 여기서 프레임을 처리
        # e.g., model.infer(frame)

        if frames_read % 10 == 0:
            elapsed = time.time() - start_time
            fps_actual = frames_read / elapsed
            print(f"  프레임 {frames_read}: 실제 처리 FPS={fps_actual:.1f}")

    cap.release()
    print(f"완료: 총 {frames_read}프레임 처리, {time.time()-start_time:.1f}초")

# 실제 IP 카메라 RTSP URL 예시:
# rtsp_url = "rtsp://admin:password@192.168.1.100:554/stream1"

# 테스트용: 로컬 웹캠을 RTSP처럼 사용
rtsp_url = 0   # 0 = 기본 웹캠
read_rtsp_stream(rtsp_url, max_frames=30)