vCodec

模块这个模块本身并没有rtsp的读写那么读写rtsp是可以使用很多方法的我们为了输出到pytorch直接使用AI程序简化rtsp

方法一

使用pipe方式也就是我们使用任何一种方式都可以如果我们有ffmpeg那么直接使用ffmpeg来读取流也是可行的使用live555

去读取流也是可行的只要把流取过来pipe给python程序就行把ffmpeg的可执行放到py文件的同一文件夹如下图所示

我们为了使用硬件解码安装了nvidia本身的PyNvCodec模块

os.name

cuda_path:os.add_dll_directory(cuda_path)else:print(CUDA_PATH

environment

os.path.isdir(path):os.add_dll_directory(path)else:print(PATH

environment

filesys.stderr)exit(1)使用ffmpeg来探测

我们可以使用ffprobe来探测我们的rtsp流用来知道流的格式是h264还是h265ok我们使用process来启动子进程来探测

def

[ffprobe,-v,quiet,-print_format,json,-show_format,-show_streams,url,]proc

json.load(bio)params

stream[height]params[framerate]

float(eval(stream[avg_frame_rate]))codec_name

True

nvc.CudaVideoCodec.HEVC)pix_fmt

pix_fmt

client实际上使用了ffmpeg的子进程并且使用管道来获取数据然后使用PyCodec来解码

def

[ffmpeg,-hide_banner,-i,url,-c:v,copy,-bsf:v,bsf_name,-f,codec_name,pipe:1,]#

Run

nvdec.DecodeSurfaceFromPacket(enc_packet,

pkt_data)if

[]urls.append(rtsp://172.28.176.1/a.264)pool

[]for

Process(targetrtsp_client,args(url,

str(uuid.uuid4()),

9),)client.start()pool.append(client)for

client

pool:client.join()我们的时间为9秒到了9秒退出程序