通过使用FFmpeg将合成的视频去抖动,达到视频的平滑展示。
import os
import cv2
import web_page
from PIL import Image
video_length = web_page.VIDEO_LENGTH
def image_to_video(image_path, media_path, fps):
'''
图片合成视频函数
:param image_path: 图片路径
:param media_path: 合成视频保存路径
:return:
'''
# 获取图片路径下面的所有图片名称
image_names = os.listdir(image_path)
# 对提取到的图片名称进行排序
image_names.sort(key=lambda n: int(n[:-4]))
# 设置写入格式
fourcc = cv2.VideoWriter_fourcc('M', 'P', '4', 'V')
# 设置每秒帧数
fps = fps
# 读取第一个图片获取大小尺寸,因为需要转换成视频的图片大小尺寸是一样的
image = Image.open(os.path.join(image_path, image_names[0]))
# 初始化媒体写入对象
media_writer = cv2.VideoWriter(media_path, fourcc, fps, image.size)
number = video_length * fps
count = 0
while True:
if count > number:
break
# 遍历图片,将每张图片加入视频当中
for image_name in image_names:
im = cv2.imread(os.path.join(image_path, image_name))
media_writer.write(im)
if count > number:
break
count += 1
for image_name in image_names[::-1]:
im = cv2.imread(os.path.join(image_path, image_name))
media_writer.write(im)
if count > number:
break
count += 1
# 释放媒体写入对象
media_writer.release()
print('无声视频写入完成!')
return media_path
video_path = image_to_video('img', 'cache/cache.mp4', 24)
# 调用视频去抖动函数
video_path = video_path
cmd = rf'ffmpeg -y -i {video_path} -vf vidstabdetect=shakiness=10:accuracy=15:result=transforms.trf -f null -'
cmd1 = rf'ffmpeg -y -i {video_path} -vf vidstabtransform=input=transforms.trf:zoom=0:smoothing=10,unsharp=5:5:0.8:3:3:0.4 {os.path.join("video", "out.mp4")}'
os.system(cmd)
os.system(cmd1)
对于命令里面的参数:
以下是关于shakiness
、accuracy
和smoothing
参数的一般范围值说明:
-
shakiness
参数:用于控制抖动检测的阈值。它的取值范围是0到100,其中0表示非常严格的抖动检测,而100表示非常宽松的抖动检测。较低的值会更严格地检测抖动,而较高的值则会更容忍抖动。你可以根据你的需求和视频的特点来调整这个值。 -
accuracy
参数:用于控制抖动检测的准确性。它的取值范围是0到15,其中0表示最低的准确性,而15表示最高的准确性。较低的值会加快抖动检测的速度,但可能会牺牲一些准确性,而较高的值会更准确地检测抖动,但可能需要更长的时间。根据你的需求和计算资源,可以选择适当的准确性级别。 -
smoothing
参数:用于控制去抖动的平滑程度。它的取值范围是0到100,其中0表示没有平滑效果,而100表示最大的平滑效果。较低的值会保留更多的原始抖动,而较高的值会更强烈地平滑抖动。你可以根据你想要的效果来调整这个值。 -
zoom参数:zoom=0
表示禁用缩放(zoom)效果。 -
unsharp
参数:unsharp
参数用于应用锐化(unsharp)效果,参数值控制锐化的强度和半径。
需要注意的是,这些参数的最佳值取决于你的具体需求、视频的特点以及个人审美偏好。建议你尝试不同的参数值,并观察输出结果,以找到最适合你的情况的参数设置。