视频防抖技术的实现

news/2024/12/13 4:54:34/

1. 介绍

视频防抖是指用于减少摄像机运动对最终视频的影响的一系列方法。摄像机的运动可以是平移(比如沿着x、y、z方向上的运动)或旋转(偏航、俯仰、翻滚)。

对视频防抖的需求在许多领域都有。

这在消费者和专业摄像中是极其重要的。因此,存在许多不同的机械、光学和算法解决方案。即使在静态图像拍摄中,防抖技术也可以帮助拍摄长时间曝光的手持照片。

在内窥镜和结肠镜等医疗诊断应用中,需要对视频进行稳定,以确定问题的确切位置和宽度。

同样,在军事应用中,无人机在侦察飞行中捕获的视频也需要进行稳定,以便定位、导航、目标跟踪等。同样的道理也适用于机器人。

2. 防抖策略

视频防抖的方法包括机械稳定方法、光学稳定方法和数字稳定方法。下面将简要讨论这些问题:

  • 机械视频稳定:机械图像稳定系统使用由特殊传感器如陀螺仪和加速度计检测到的运动来移动图像传感器以补偿摄像机的运动。
  • 光学视频稳定:在这种方法中,不是移动整个摄像机,而是通过镜头的移动部分来实现稳定。这种方法使用了一个可移动的镜头组合,当光通过相机的镜头系统时,可以可变地调整光的路径长度。
  • 数字视频稳定:这种方法不需要特殊的传感器来估计摄像机的运动。主要有三个步骤:1)运动估计2)运动平滑,3)图像合成。第一步导出了两个连续坐标系之间的变换参数。第二步过滤不需要的运动,在最后一步重建稳定的视频。

在这篇文章中,我们将学习一个快速和鲁棒性好的数字视频稳定算法的实现。它是基于二维运动模型,其中我们应用欧几里得(即相似性)变换包含平移、旋转和缩放。

在这里插入图片描述
正如你在上面的图片中看到的,在欧几里得运动模型中,图像中的一个正方形可以转换为任何其他位置、大小或旋转不同的正方形。它比仿射变换和单应变换限制更严格,但对于运动稳定来说足够了,因为摄像机在视频连续帧之间的运动通常很小。

3. 基于点特征匹配的视频防抖

该方法涉及跟踪两个连续帧之间的多个特征点。跟踪特征允许我们估计帧之间的运动并对其进行补偿。

下面的流程图显示了基本步骤。
在这里插入图片描述

3.1 设置输入和输出视频

首先,让我们完成读取输入视频和写入输出视频的设置。

# Import numpy and OpenCV
import numpy as np
import cv2# Read input video
cap = cv2.VideoCapture('video.mp4') # Get frame count
n_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) # Get width and height of video stream
w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) 
h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))# Define the codec for output video
fourcc = cv2.VideoWriter_fourcc(*'MJPG')# Set up output video
out = cv2.VideoWriter('video_out.mp4', fourcc, fps, (w, h))

3.2 读取第一帧并将其转换为灰度图

对于视频稳定,我们需要捕捉视频的两帧,估计帧之间的运动,最后校正运动。

# Read first frame
_, prev = cap.read() # Convert frame to grayscale
prev_gray = cv2.cvtColor(prev, cv2.COLOR_BGR2GRAY)

3.3 寻找帧之间的移动

这是算法中最关键的部分。我们将遍历所有的帧,并找到当前帧和前一帧之间的移动。没有必要知道每一个像素的运动。欧几里得运动模型要求我们知道两个坐标系中两个点的运动。但是在实际应用中,找到50-100个点的运动,然后用它们来稳健地估计运动模型是一个好方法。

  1. 可用于跟踪的优质特征
    现在的问题是我们应该选择哪些点进行跟踪。请记住,跟踪算法使用一个小补丁围绕一个点来跟踪它。这样的跟踪算法受到孔径问题的困扰,如下面的视频所述, 因此,光滑的区域不利于跟踪,而有很多角的纹理区域则比较好。幸运的是,OpenCV有一个快速的特征检测器,可以检测最适合跟踪的特性。它被称为goodFeaturesToTrack)
  2. Lucas-Kanade光流
    一旦我们在前一帧中找到好的特征,我们就可以使用Lucas-Kanade光流算法在下一帧中跟踪它们。
    它是利用OpenCV中的calcOpticalFlowPyrLK函数实现的。在calcOpticalFlowPyrLK这个名字中,LK代表Lucas-Kanade,而Pyr代表金字塔。计算机视觉中的图像金字塔是用来处理不同尺度(分辨率)的图像的。
    由于各种原因,calcOpticalFlowPyrLK可能无法计算出所有点的运动。例如,当前帧的特征点可能会被下一帧的另一个对象遮挡。幸运的是,您将在下面的代码中看到,calcOpticalFlowPyrLK中的状态标志可以用来过滤掉这些值。
  3. 估计运动
    在步骤1中,我们在前一帧中找到了一些好的特征。在步骤2中,我们使用光流来跟踪特征。换句话说,我们已经找到了特征在当前帧中的位置,并且我们已经知道了特征在前一帧中的位置。所以我们可以使用这两组点来找到映射前一个坐标系到当前坐标系的刚性(欧几里德)变换。这是使用函数estimateRigidTransform完成的。
    一旦我们估计了运动,我们可以把它分解成x和y的平移和旋转(角度)。我们将这些值存储在一个数组中,这样就可以平稳地更改它们。

下面的代码将完成步骤1到3。

# Pre-define transformation-store array
transforms = np.zeros((n_frames-1, 3), np.float32) for i in range(n_frames-2):# Detect feature points in previous frameprev_pts = cv2.goodFeaturesToTrack(prev_gray,maxCorners=200,qualityLevel=0.01,minDistance=30,blockSize=3)# Read next framesuccess, curr = cap.read() if not success: break # Convert to grayscalecurr_gray = cv2.cvtColor(curr, cv2.COLOR_BGR2GRAY) # Calculate optical flow (i.e. track feature points)curr_pts, status, err = cv2.calcOpticalFlowPyrLK(prev_gray, curr_gray, prev_pts, None) # Sanity checkassert prev_pts.shape == curr_pts.shape # Filter only valid pointsidx = np.where(status==1)[0]prev_pts = prev_pts[idx]curr_pts = curr_pts[idx]#Find transformation matrixm = cv2.estimateRigidTransform(prev_pts, curr_pts, fullAffine=False) #will only work with OpenCV-3 or less# Extract traslationdx = m[0,2]dy = m[1,2]# Extract rotation angleda = np.arctan2(m[1,0], m[0,0])# Store transformationtransforms[i] = [dx,dy,da]# Move to next frameprev_gray = curr_grayprint("Frame: " + str(i) +  "/" + str(n_frames) + " -  Tracked points : " + str(len(prev_pts)))

3.4 计算帧之间的平滑运动

在前面的步骤中,我们估计帧之间的运动并将它们存储在一个数组中。我们现在需要通过叠加上一步估计的微分运动来找到运动轨迹。

  1. 轨迹计算
    在这一步,我们将增加运动之间的帧来计算轨迹。我们的最终目标是平滑这条轨迹。
    Python 在Python中,可以很容易地使用numpy中的cumsum(累计和)来实现。
# Compute trajectory using cumulative sum of transformations
trajectory = np.cumsum(transforms, axis=0
  1. 计算平滑轨迹
    在上一步中,我们计算了运动轨迹。所以我们有三条曲线来显示运动(x, y,和角度)如何随时间变化。
    在这一步,我们将展示如何平滑这三条曲线。
    平滑任何曲线最简单的方法是使用移动平均滤波器(moving average filter)。顾名思义,移动平均过滤器将函数在某一点上的值替换为由窗口定义的其相邻函数的平均值。让我们看一个例子。
    假设我们在数组c中存储了一条曲线,那么曲线上的点是c[0]…c[n-1]。设f是我们通过宽度为5的移动平均滤波器过滤c得到的平滑曲线。
    该曲线的 k t h k^{th} kth元素使用
    在这里插入图片描述
    如您所见,平滑曲线的值是噪声曲线在一个小窗口上的平均值。下图显示了左边的噪点曲线的例子,使用右边的尺度为5 滤波器进行平滑。
    在这里插入图片描述
    在Python实现中,我们定义了一个移动平均滤波器,它接受任何曲线(即1-D的数字)作为输入,并返回曲线的平滑版本。
def movingAverage(curve, radius): window_size = 2 * radius + 1# Define the filter f = np.ones(window_size)/window_size # Add padding to the boundaries curve_pad = np.lib.pad(curve, (radius, radius), 'edge') # Apply convolution curve_smoothed = np.convolve(curve_pad, f, mode='same') # Remove padding curve_smoothed = curve_smoothed[radius:-radius]# return smoothed curvereturn curve_smoothed

我们还定义了一个函数,它接受轨迹并对这三个部分进行平滑处理。

def smooth(trajectory): smoothed_trajectory = np.copy(trajectory) # Filter the x, y and angle curvesfor i in range(3):smoothed_trajectory[:,i] = movingAverage(trajectory[:,i], radius=SMOOTHING_RADIUS)return smoothed_trajectory

最后使用

# Compute trajectory using cumulative sum of transformations
trajectory = np.cumsum(transforms, axis=0)
  1. 计算平滑变换
    到目前为止,我们已经得到了一个平滑的轨迹。在这一步,我们将使用平滑的轨迹来获得平滑的变换,可以应用到视频的帧来稳定它。
    这是通过找到平滑轨迹和原始轨迹之间的差异,并将这些差异加回到原始的变换中来完成的。
# Calculate difference in smoothed_trajectory and trajectory
difference = smoothed_trajectory - trajectory# Calculate newer transformation array
transforms_smooth = transforms + difference

5. 将平滑的摄像机运动应用到帧中

差不多做完了。现在我们所需要做的就是循环帧并应用我们刚刚计算的变换。

如果我们有一个指定为(x, y, θ \theta θ),的运动,对应的变换矩阵是

在这里插入图片描述

# Reset stream to first frame 
cap.set(cv2.CAP_PROP_POS_FRAMES, 0) # Write n_frames-1 transformed frames
for i in range(n_frames-2):# Read next framesuccess, frame = cap.read() if not success:break# Extract transformations from the new transformation arraydx = transforms_smooth[i,0]dy = transforms_smooth[i,1]da = transforms_smooth[i,2]# Reconstruct transformation matrix accordingly to new valuesm = np.zeros((2,3), np.float32)m[0,0] = np.cos(da)m[0,1] = -np.sin(da)m[1,0] = np.sin(da)m[1,1] = np.cos(da)m[0,2] = dxm[1,2] = dy# Apply affine wrapping to the given frameframe_stabilized = cv2.warpAffine(frame, m, (w,h))# Fix border artifactsframe_stabilized = fixBorder(frame_stabilized) # Write the frame to the fileframe_out = cv2.hconcat([frame, frame_stabilized])# If the image is too big, resize it.if(frame_out.shape[1] > 1920): frame_out = cv2.resize(frame_out, (frame_out.shape[1]/2, frame_out.shape[0]/2));cv2.imshow("Before and After", frame_out)cv2.waitKey(10)out.write(frame_out)
  1. 修复边界伪影
    当我们稳定一个视频,我们可能会看到一些黑色的边界伪影。这是意料之中的,因为为了稳定视频,帧可能不得不缩小大小。
    我们可以通过将视频的中心缩小一小部分(例如4%)来缓解这个问题。
    下面的fixBorder函数显示了实现。我们使用getRotationMatrix2D,因为它在不移动图像中心的情况下缩放和旋转图像。我们所需要做的就是调用这个函数时,旋转为0,缩放为1.04(也就是提升4%)。
def fixBorder(frame):s = frame.shape# Scale the image 4% without moving the centerT = cv2.getRotationMatrix2D((s[1]/2, s[0]/2), 0, 1.04)frame = cv2.warpAffine(frame, T, (s[1], s[0]))return frame

5. 结果

本文分享的视频防抖代码的结果如上所示。我们的目标是显著减少运动,但不是完全消除它。

目前的方法只适用于固定长度的视频,而不适用于实时feed。我不得不对这个方法进行大量修改,以获得实时视频输出,这超出了本文的范围,但这是可以实现的,更多的信息可以在这里找到。

https://abhitronix.github.io/2018/11/30/humanoid-AEAM-3/

6. 优缺点

6.1 优点

这种方法对低频运动(较慢的振动)具有良好的稳定性。这种方法内存消耗低,因此非常适合嵌入式设备(如树莓派)。这种方法对视频缩放抖动有很好的效果。

6.2 缺点

这种方法对高频扰动的抵抗效果很差。如果有一个严重的运动模糊,特征跟踪将失败,结果将不是最佳的。这种方法也不适用于滚动快门失真。

7. 参考文献

  • Example video and Code reference from Nghia Ho’s post
  • https://www.learnopencv.com/video-stabilization-using-point-feature-matching-in-opencv/

8. 最后


http://www.ppmy.cn/news/480336.html

相关文章

EIS(电子稳像,视频防抖)算法开发记录

最近项目中有个小需求,需要对运动视频使用电子稳像算法进行稳像处理。开发从0开始,在此记录一下开发过程和心得,做个备份,期间也走了很多弯路,供大家参考。最终效果个人认为基本可媲美主流手机。先上最终效果&#xff…

光学防抖OIS介绍与标定

1.开篇直接先看一下有无OIS的效果差异 手持设备拍照时,手的抖动(点击抖动和生理抖动)会造成相机的轻微倾斜(一般在+/-0.5度以内),该倾斜引起了镜头观察角度的变化,以镜头为参照物来说,相当于被拍摄的物体移动了,因此所成的像也会在图像传感器上相对于原位置发生偏移, 结果造成图…

Camera:多帧降噪及photo、video模式下的防抖处理

今天突然遇到手机相机拍出的照片没有多帧降噪效果了,Tunning很多的效果调试需要基于多帧降噪进行处理,最后很坑爹的发现多帧降噪居然与EIS、AIS等防抖的处理有关系;带着满脸的黑人问号研究了一下相机的多帧降噪以及AIS、EIS的防抖处理。 1.多…

mysql时间字段根据当前时间更新

保存数据时默认值设置为当前时间 create_time datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT 创建时间;保存行数据时默认值设置为当前时间,同时更新字段时自动更新为当前时间 update_time datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_T…

Vue 点击添加className

例如如下dom <li class"del" click"del($event)">删除</li>因为js获取兄弟元素需要中转&#xff0c;所以先获取父元素&#xff0c;再获取子元素&#xff0c;移除className&#xff0c;再给当前dom添加className const siblings e.currentTa…

Vue脚手架

目录 一、脚手架文件结构1.1 创建脚手架 二、关于不同版本的Vue三、vue.config.js配置文件四、ref属性五、props配置项六、mixin(混入)七、插件八、scoped样式九、小结十、webStorage十一、组件的自定义事件十二、全局事件总线&#xff08;GlobalEventBus&#xff09;十三、nex…

win10计算机怎样双屏,有什么办法让window10双屏显示_教你一招让window10双屏的方法...

win10电脑运行多个程序会不知不觉打开多个不同类型的窗口&#xff0c;窗口多了总要时不时的来回频繁切换&#xff0c;总感觉很麻烦。如果不想频繁切换&#xff0c;我们可以让window10双屏显示&#xff0c;而且win10系统双屏显示设置步骤并不困难&#xff0c;下面小编就为大家演…

Linux vi 双屏显示,如何设置双屏显示器

大家好&#xff0c;我是时间财富网智能客服时间君&#xff0c;上述问题将由我为大家进行解答。 以电脑为例&#xff0c;其设置双屏显示器的方法是&#xff1a; 1、首先要准备好必要的硬件设备&#xff0c;主机一台&#xff0c;显示器两个&#xff0c;vga线三根&#xff0c;分屏…