source_rtmp="rtmp://xxx/live/7CTDLB100A0023-165-0-7"
logger.info("识别数据源："+source_rtmp)
cv2.VideoCapture(source_rtmp)     

while True:
    ret, frame = cap.read()
    
    if not ret:
        print("Error reading frame from RTMP stream!")
        break

import ffmpeg

def get_source_info_ffmpeg(source_name):
    return_value = 0
    try:
        info = ffmpeg.probe(source_name)
        json_str = json.dumps(info)
        print(json_str)
        # print("---------------------------------")
        vs = next(c for c in info['streams'] if c['codec_type'] == 'video')
        format_name = info['format']['format_name']
        codec_name = vs['codec_name']
        # duration_ts = float(vs['duration_ts'])
        fps = vs['r_frame_rate']
        width = vs['width']
        height = vs['height']
        print("format_name:{} \ncodec_name:{} \nwidth:{} \nheight:{} \nfps:{}".format(format_name, codec_name, width, height, fps))
    except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
        print("init_source:{} error. {}\n".format(source_name, str(e)))
        return_value = 0
    return return_value

def main():
    source_name = "rtmp://openmap.tech:10009/live/1581F6Q8D242S00CC0NW-81-0-0"
    get_source_info_ffmpeg(source_name)

if __name__ == "__main__":
    main()      

import ffmpeg
# 创建RTMP输入格式对象
stream = ffmpeg.input(rtmp_url).output(output_file, f='flv')
# 开始拉流并保存为文件
ffmpeg.run(stream)

import av
from PIL import Image
import numpy as np


def process_frame(frame):
    # 将帧转换为numpy数组
    img_array = frame.to_ndarray()

    # 转换为PIL图像
    img = Image.fromarray(img_array)

    # 保存图像
    img.save(f"frame_{frame.index}.png") 


    # 连接RTMP流
rtmp_url = "rtmp://47.97.116.4:10009/live/1581F6Q8D23CJ00A420Y-81-0-0"
input_container = av.open(rtmp_url)

# 选择视频流
video_stream = input_container.streams.video[0]

# 读取视频帧
for frame in input_container.decode(video_stream):
    # 处理每一帧
    process_frame(frame)

import subprocess as sp
import numpy as np

FFMPEG_BIN = "ffmpeg"

command2 = [ FFMPEG_BIN,
        '-y', # (optional) overwrite output file if it exists
        '-f', 'rawvideo',     #-f fmt 强迫采用格式fmt
        '-vcodec','rawvideo',
        '-s', '960x540', # size of one frame
        '-pix_fmt', 'rgb24',
        '-r', '24', # frames per second
        '-i', '-', # The imput comes from a pipe
        '-an', # Tells FFMPEG not to expect any audio
        #'-vcodec', 'libx264',
        'rtmp://openmap.tech:10009/live/1581F6Q8D241Q00C32EE-81-0-0' ]

proc2stream = sp.Popen( command2, stdin=sp.PIPE,bufsize=10**8)

command1 = [FFMPEG_BIN,
           '-i', 'rtmp://openmap.tech:10009/live/1581F6Q8D241Q00C32EE-81-0-01' , 
           '-f', 'image2pipe', 
           '-pix_fmt', 'rgb24',
           '-vcodec', 'rawvideo', '-']

proc2pics = sp.Popen(command1,stdout=sp.PIPE,bufsize=10**8)

while True:
    raw_image = proc2pics.stdout.read(960*540*3)
    image = np.fromstring(raw_image,dtype='uint8')
    #print(type(image))
    #image = image.reshape((540,960,3))
    #print(image.shape)
    # proc2stream.stdin.write(image.tostring())

import cv2

# RTMP视频流地址
rtmp_url = "rtmp://xx:10009/live/xxx-81-0-0"
# 创建VideoCapture对象
cap = cv2.VideoCapture(rtmp_url)
# 检查是否成功打开视频流
if not cap.isOpened():
    print("无法打开视频流")
    exit()

i=0
while True:
    # 读取视频帧
    ret, frame = cap.read()
    # 检查是否成功读取到帧
    if not ret:
        # print("无法获取视频帧")
        continue
    i=i+1
    print(i)
    # 显示图像
    cv2.imshow("RTMP Video", frame)
    # 按下 'q' 键退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
# 释放资源
cap.release()
cv2.destroyAllWindows()