import json import time import subprocess as sp import ffmpeg import cv2 import sys import numpy as np """ 获取视频基本信息 """ def get_video_info(in_file): try: probe = ffmpeg.probe('https://vod.play.t-aaron.com/customerTrans/edc96ea2115a0723a003730956208134/55547af9-184f0827dae-0004-f90c-f2c-7ec68.mp4') # format = probe['format'] # size = int(format['size'])/1024/1024 video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None) if video_stream is None: print('No video stream found', file=sys.stderr) return width = int(video_stream['width']) height = int(video_stream['height']) num_frames = int(video_stream['nb_frames']) up, down = str(video_stream['r_frame_rate']).split('/') fps = eval(up) / eval(down) print("fbs:", fps) # duration = float(video_stream['duration']) bit_rate = int(video_stream['bit_rate'])/1000 print('width: {}'.format(width)) print('height: {}'.format(height)) # print('num_frames: {}'.format(num_frames)) print('bit_rate: {}k'.format(bit_rate)) # print('fps: {}'.format(fps)) # print('size: {}MB'.format(size)) # print('duration: {}'.format(duration)) return video_stream except Exception as err: if isinstance(err, ffmpeg._run.Error): print(err.stderr.decode(encoding = 'utf-8')) raise err if __name__ == '__main__': file_path = 'https://vod.play.t-aaron.com/customerTrans/edc96ea2115a0723a003730956208134/55547af9-184f0827dae-0004-f90c-f2c-7ec68.mp4' #file_path = 'https://vod.play.t-aaron.com/customerTrans/edc96ea2115a0723a003730956208134/40b416f7-183b57f6be0-0004-f90c-f2c-7ec68.mp4' #file_path = 'https://vod.play.t-aaron.com/3301fc8e166f45be88f2214e7a8f4a9d/e29535365b54434d9ed2e8c3b0a175da-fba35541b31a1049ca05b145a283c33a-hd.mp4' video_info = get_video_info(file_path) print(json.dumps(video_info)) # total_frames = int(video_info['nb_frames']) # print('总帧数：' + str(total_frames)) # random_frame = random.randint(1, total_frames) # print('随机帧：' + str(random_frame)) # out = read_frame_as_jpeg(file_path, i) # image_array = numpy.asarray(bytearray(out), dtype="uint8") # image = cv2.imdecode(image_array, cv2.IMREAD_COLOR) # kwargs={'fflags': 'nobuffer', 'flags': 'low_delay'} kwargs={ # "hwaccel": "nvdec", # "vcodec": "h264_cuvid", # "c:v": "h264_cuvid" } output_args = { # "vcodec": "hevc_nvenc", # "c:v": "hevc_nvenc", # "preset": "fast", } # i = 1 # process1 = ( # ffmpeg # .input(file_path, **kwargs) # .output('pipe:', format='rawvideo', pix_fmt='rgb24', **output_args) # # .global_args("-an") # .overwrite_output() # .run_async(pipe_stdout=True, pipe_stderr=True) # ) width = int(video_info['width']) height = int(video_info['height']) width_2_1 = int(width/2) height_2_1 = int(height/2) print("长:", width, "宽:", height) command = ['ffmpeg', # '-hwaccel', 'cuvid', '-c:v', 'h264_cuvid', # '-hwaccel_output_format', 'cuda', '-resize', '%sx%s' % (width_2_1, height_2_1), '-i', file_path, # '-c:v', 'hevc_nvenc', # '-pix_fmt', 'yuv420p', '-f', 'rawvideo', # '-pix_fmt', 'bgr24', '-an', '-'] p = sp.Popen(command, stdout=sp.PIPE) # ai_video_file = cv2.VideoWriter(r"C:\Users\chenyukun\Desktop\shipin\aa.mp4", cv2.VideoWriter_fourcc(*'mp4v'), 30, # (width, height)) command1 = ['ffmpeg', # '-loglevel', 'debug', '-y', # 不经过确认，输出时直接覆盖同名文件。 '-f', 'rawvideo', '-vcodec', 'rawvideo', '-pix_fmt', 'bgr24', # '-hwaccel', 'cuvid', # '-c:v', 'h264_cuvid', # '-hwaccel_output_format', 'cuda', # '-s', "{}x{}".format(self.width * 2, self.height), '-s', "{}x{}".format(width_2_1, height_2_1), '-r', str(25), '-i', '-', # 指定输入文件 '-g', str(5), # '-maxrate', '15000k', # '-profile:v', 'high', # '-b:v', '4000k', # '-crf', '18', '-rc:v', 'vbr', '-cq:v', '30', '-qmin', '30', '-qmax', '30', '-bufsize', '1', '-c:v', 'h264_nvenc', # # '-c:v', 'libx264', # 指定视频编码器 # '-tune', 'zerolatency', # 加速编码速度 # '-sc_threshold', '0', '-pix_fmt', 'yuv420p', "-an", # '-flvflags', 'no_duration_filesize', '-preset', 'fast', # '-f', 'flv',ags', 'no_duration_filesize', # '-preset', 'fast', # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast, # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。 '-f', 'flv', 'rtmp://live.push.t-aaron.com/live/THSAr'] # # 管道配置 p1 = sp.Popen(command1, stdin=sp.PIPE) start1 = time.time() num = 0 while True: num += 1 # if num ==100: # print(time.time()-start1) # break start = time.time() in_bytes = p.stdout.read(int(width * height*3//8)) # print(type(in_bytes)) # img = (np.frombuffer(in_bytes, np.uint8)).reshape((height*3//4, width//2)) # result = cv2.cvtColor(img, cv2.COLOR_YUV2BGR_NV12) # result = cv2.resize(result, (int(width / 2), int(height / 2)), interpolation=cv2.INTER_LINEAR) # result = cv2.cvtColor(img, cv2.COLOR_RGB2BGR) if not in_bytes: print(in_bytes) # ai_video_file.release() p.stdout.close() p.wait() break print(num) time.sleep(0.001) # p1.stdin.write(result.tostring()) # p1.stdin.write(in_frame.tostring()) # frame # .astype(np.uint8) # .tobytes() # frame = cv2.resize(in_frame, (1280, 720)) # 改变图片尺寸 # frame = cv2.cvtColor(in_frame, cv2.COLOR_RGB2BGR) # 转成BGR # i += 1 # print(round(time.time()-start, 5)) # # ai_video_file.write(in_frame) # if time.time() - start1 > 60: # ai_video_file.release() # p.stdout.close() # p.wait() # break # cv2.imshow('frame', frame) # time.sleep(1111) p.kill()