175 lines
6.3 KiB
Python
175 lines
6.3 KiB
Python
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 = r'D:\shipin\777.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'}
|
|
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)
|
|
text = ''
|
|
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)
|
|
# for line in p.stderr:
|
|
# # print(line.strip())
|
|
# for line in p.stdout:
|
|
# print(text)
|
|
|
|
command1 = ['ffmpeg',
|
|
'-report',
|
|
# '-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, height),
|
|
'-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, stdout=sp.PIPE, stderr=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))
|
|
# # print(type(in_bytes))
|
|
img = (np.frombuffer(in_bytes, np.uint8)).reshape((height, width, 3))
|
|
# # 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)
|
|
|
|
for line in iter(p1.stdout.readline, b''):
|
|
print(line)
|
|
# if in_bytes:
|
|
# p1.stdin.write(img.tostring())
|
|
# print(p1.stderr.read().decode('utf-8'))
|
|
# for line in iter(p1.std
|
|
#
|
|
#
|
|
# out.readline, ''):
|
|
# line = line.decode('utf-8')
|
|
# print(line)
|
|
# print(num)
|
|
# time.sleep(0.001)
|
|
|
|
# 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() |