[OpenCV实战]6 基于特征点匹配的视频稳像

在这篇文章中,我们将学习如何使用OpenCV库中称为特征点匹配的技术以实现简单视频稳定稳像。我们将讨论该算法并共享代码,以便在OpenCV中使用此方法设计一个简单的稳定器,最好OpenCV3.4.3以上实现代码。什么是视频稳定,视频稳定是指用于减少相机运动对最终视频影响的一系列方法,理解成消除视频抖动就行了。见下图通常用拍摄会出现轻微的抖动,比如手机拍摄视频,后期需要对其进行视频稳像操作。

视频稳定的需求涉及许多领域。它在消费者和专业摄像中极为重要。因此,存在许多不同的机械,光学和算法解决方案。即使在静态图像拍摄中,稳定也可以帮助拍摄时间长的照片。在内窥镜检查和结肠镜检查等医疗诊断应用中,需要稳定视频以确定问题的确切位置和宽度。类似地,在军事应用中,飞行器在侦察飞行中捕获的视频需要稳定以进行定位,导航,目标跟踪等。这同样适用于机器人应用。

介绍

1.1 视频稳定的方法

视频稳定方法包括机械,光学和数字稳定方法。具体如下:

机械视频稳定:机械图像稳定系统使用陀螺仪和加速度计等特殊传感器检测到运动来移动图像传感器以补偿相机的运动。

光学视频稳定:在这种方法中,不是移动整个相机,而是通过移动镜头的部分来实现稳定。该方法采用可移动透镜组件,当透过相机的透镜系统时,可移动透镜组件可变地调节光的路径长度。

数字视频稳定:此方法不需要特殊的传感器来估算相机运动。本文就是用的这种方法。主要有三个步骤, 1)运动估计,2)运动平滑,3)图像合成。在第一阶段中导出两个连续帧之间的变换参数。第二级滤除不需要的运动,第三个阶段重建稳定的视频。

我们将在本文中学习快速而强大的实数字视频稳定算法。它基于二维运动模型,我们应用包含平移,旋转和缩放的欧几里德(又称相似性)变换。

如上图所示,在欧几里德运动模型中,图像中的正方形可以转换为具有不同大小,形状位置的任何其他四方形。它比仿射和单应变换更具限制性,但足以用于运动稳定,因为视频的连续帧之间的相机移动通常很小。

1.2 使用点特征匹配的视频稳定

该方法涉及跟踪两个连续帧之间的一些特征点。跟踪的特征允许我们估计帧之间的运动并对其进行补偿。下面的流程图显示了算法基本步骤。

1 获取多帧视频图像,获取图像角点(特征点);

2 光流法跟踪角点;根据前后两张图像角点变化得到表示运动的仿射变化矩阵。

3 根据仿射变化矩阵计算运动轨迹,并且平滑运动轨迹。

4 根据平滑后的运动轨迹,得到平滑运动后的仿射变化矩阵。

5 根据平滑运动后的仿射变化矩阵得到稳定后的图像。

 

https://www.learnopencv.com/wp-content/uploads/2019/01/AEAM-3.png

2 算法

2.1 帧间运动信息获取

算法中最关键的部分是确定各帧的运动方向。我们将迭代所有帧,并找到当前帧和前一帧之间的运动。没有必要知道每个像素的运动。欧几里德运动模型要求我们知道两帧中2个点的运动信息。然而,在实践中最好找到50-100点的运动信息,然后使用它们来稳健地估计运动模型

2.1.1 合适的特征点获取

现在的问题是我们应该选择哪些特征点进行跟踪。请记住,跟踪算法会使用一个以该点为圆心的圆(小孔)来近似来模拟点的运动。这种跟踪算法受到圆直径的影响。因此,平滑区域对于跟踪是不利的,并且具有许多角点的纹理区域是好的。幸运的是,OpenCV具有快速检测函数特征点的跟踪,即函数goodFeaturesToTrack。角点个人理解是指图像中亮度变化剧烈的点或图像边缘上变化交大的点。

2.1.2 Lucas-Kanade光流法

一旦我们在前一帧中找到了好的特征(角点),我们就可以使用名为Lucas-Kanade Optical Flow的算法在下一帧中跟踪它们,该算法以算法的发明者命名。算法详情可以见

https://blog.csdn.net/linmingan/article/details/79296963

OpenCV中的函数calcOpticalFlowPyrLK实现算法。在calcOpticalFlowPyrLK中,LK代表Lucas-Kanade,而Pyr代表金字塔。计算机视觉中的图像金字塔用于处理不同比例(分辨率)的图像。但是由于各种原因,calcOpticalFlowPyrLK可能无法计算所有点的运动。例如,当前帧中的特征点可能被下一帧中的另一个对象遮挡。幸运的是calcOpticalFlowPyrLK中的状态标志可用于过滤掉这些值。

2.1.3 运动估计

回顾一下,在步骤2.1.1中,我们发现在前一帧中要跟踪的特征点。在步骤2.1.2中,我们使用光流来跟踪特征点。换句话说,我们在当前帧中找到了特征点的位置,并且我们已经知道了前一帧中特征点的位置。因此,我们可以使用这两组特征点来找到将前一帧映射到当前帧的(欧几里德)变换。OpenCV使用函数estimateRigidTransform完成的。仿射变化详细见

https://blog.csdn.net/dongfang1984/article/details/52959308

一旦我们获取运动信息,我们就可以将它分解为x和y值以及平移和旋转(角度)值。我们将这些值存储在一个数组中,以便我们可以顺利更改它们。

下面的代码将介绍步骤2.1.1至2.1.3。请务必阅读要遵循的代码中的注释。

在C ++实现中,我们首先定义一些类来帮助我们存储估计的运动矢量。下面C++代码的TransformParam类存储运动信息(dx为x轴运动信息,dy为y轴运动信息,da为角度信息),并提供方法getTransform将此运动转换为变换矩阵。

 
  1. /**

  2. * @brief 运动信息结构体

  3. *

  4. */

  5. struct TransformParam

  6. {

  7. TransformParam() {}

  8. //x轴信息,y轴信息,角度信息

  9. TransformParam(double _dx, double _dy, double _da)

  10. {

  11. dx = _dx;

  12. dy = _dy;

  13. da = _da;

  14. }

  15.  
  16. double dx;

  17. double dy;

  18. // angle

  19. double da;

  20.  
  21. void getTransform(Mat &T)

  22. {

  23. // Reconstruct transformation matrix accordingly to new values 重建变换矩阵

  24. T.at<double>(0, 0) = cos(da);

  25. T.at<double>(0, 1) = -sin(da);

  26. T.at<double>(1, 0) = sin(da);

  27. T.at<double>(1, 1) = cos(da);

  28.  
  29. T.at<double>(0, 2) = dx;

  30. T.at<double>(1, 2) = dy;

  31. }

  32. };

我们循环遍历帧并执行2.1帧间运动信息获取所有代码。C++代码:

 
  1. //previous transformation matrix 上一张图像的仿射矩阵

  2. Mat last_T;

  3. //从第二帧开始循环遍历视频所有帧

  4. for (int i = 1; i < n_frames; i++)

  5. {

  6. // Vector from previous and current feature points 前一帧角点vector,当前帧角点vector

  7. vector<Point2f> prev_pts, curr_pts;

  8.  
  9. // Detect features in previous frame 获取前一帧的角点

  10. //前一帧灰度图,前一帧角点vector, 最大角点数,检测到的角点的质量等级,两个角点之间的最小距离

  11. goodFeaturesToTrack(prev_gray, prev_pts, 200, 0.01, 30);

  12.  
  13. // Read next frame 读取当前帧图像

  14. bool success = cap.read(curr);

  15. if (!success)

  16. {

  17. break;

  18. }

  19.  
  20. // Convert to grayscale 将当前帧图像转换为灰度图

  21. cvtColor(curr, curr_gray, COLOR_BGR2GRAY);

  22.  
  23. // Calculate optical flow (i.e. track feature points) 光流法追寻特征点

  24. //输出状态矢量(元素是无符号char类型,uchar),如果在当前帧发现前一帧角点特征则置为1,否则,为0

  25. vector<uchar> status;

  26. //输出误差矢量

  27. vector<float> err;

  28. //光流跟踪

  29. //前一帧灰度图像,当前帧灰度图像,前一帧角点,当前帧角点,状态量,误差量

  30. calcOpticalFlowPyrLK(prev_gray, curr_gray, prev_pts, curr_pts, status, err);

  31.  
  32. // Filter only valid points 获取光流跟踪下有效的角点

  33. //遍历角点

  34. auto prev_it = prev_pts.begin();

  35. auto curr_it = curr_pts.begin();

  36. for (size_t k = 0; k < status.size(); k++)

  37. {

  38. if (status[k])

  39. {

  40. prev_it++;

  41. curr_it++;

  42. }

  43. //删除无效角点

  44. else

  45. {

  46. prev_it = prev_pts.erase(prev_it);

  47. curr_it = curr_pts.erase(curr_it);

  48. }

  49. }

  50.  
  51. // Find transformation matrix 获得变换矩阵

  52. //false表示带几何约束的仿射变换,true则是全仿射变化,T为变换矩阵

  53. Mat T = estimateRigidTransform(prev_pts, curr_pts, false);

  54.  
  55. // In rare cases no transform is found.

  56. // We'll just use the last known good transform.

  57. //极少数情况会找不到变换矩阵,取上一个变换为当前变化矩阵

  58. //当然第一次检测就没找到仿射矩阵,算法会出问题,不过概率很低

  59. if (T.data == NULL)

  60. {

  61. last_T.copyTo(T);

  62. }

  63. T.copyTo(last_T);

  64.  
  65. // Extract traslation 提取仿射变化结果

  66. double dx = T.at<double>(0, 2);

  67. double dy = T.at<double>(1, 2);

  68.  
  69. // Extract rotation angle 提取角度

  70. double da = atan2(T.at<double>(1, 0), T.at<double>(0, 0));

  71.  
  72. // Store transformation 存储仿射变化矩阵

  73. transforms.push_back(TransformParam(dx, dy, da));

  74.  
  75. // Move to next frame 进行下一次检测准测

  76. curr_gray.copyTo(prev_gray);

  77.  
  78. cout << "Frame: " << i << "/" << n_frames << " - Tracked points : " << prev_pts.size() << endl;

  79. }

2.2 计算帧之间的总体运动

在上一步中,我们获取了帧之间的运动情况并将它们存储在一个数组中。我们现在需要通过累积分析帧间运动情况来找到运动轨迹。

2.2.1 计算运动轨迹

在此步骤中,我们将累加帧之间的运动以计算轨迹。我们的最终目标是平滑这一轨迹。

在Python中,使用numpy中的cumsum(累积和)很容易实现。

在C ++中,我们定义了一个名为Trajectory的类来存储运动参数每次的累积和。

 
  1. /**

  2. * @brief 轨迹结构体

  3. *

  4. */

  5. struct Trajectory

  6. {

  7. Trajectory() {}

  8. Trajectory(double _x, double _y, double _a)

  9. {

  10. x = _x;

  11. y = _y;

  12. a = _a;

  13. }

  14.  
  15. double x;

  16. double y;

  17. // angle

  18. double a;

  19. };

我们还定义了一个函数cumsum,输入为TransformParams结构数据,并通过dx,dy和da(角度)的累积和来返回轨迹信息。C++代码:

 
  1. /**

  2. * @brief 轨迹累积

  3. *

  4. * @param transforms 运动信息结构体

  5. * @return vector<Trajectory> 轨迹结构体

  6. */

  7. vector<Trajectory> cumsum(vector<TransformParam> &transforms)

  8. {

  9. // trajectory at all frames 所有帧的运动轨迹

  10. vector<Trajectory> trajectory;

  11. // Accumulated frame to frame transform 累加计算x,y以及a(角度)

  12. double a = 0;

  13. double x = 0;

  14. double y = 0;

  15.  
  16. //累加

  17. for (size_t i = 0; i < transforms.size(); i++)

  18. {

  19. x += transforms[i].dx;

  20. y += transforms[i].dy;

  21. a += transforms[i].da;

  22.  
  23. trajectory.push_back(Trajectory(x, y, a));

  24. }

  25.  
  26. return trajectory;

  27. }

2.2.2 计算平滑轨迹

在上一步中,我们计算了运动的轨迹。因此,我们有三条曲线显示运动(x,y和角度)随时间的变化情况。

在这一步中,我们将展示如何平滑这三条曲线。

平滑任何曲线的最简单方法是使用移动平均滤波器。顾名思义,移动平均滤波器将该点处的函数值替换为邻域窗格所有点的平均值。我们来看一个例子。

比方说,我们已经存储在数组中的曲线C,曲线上的点为C [0], ... ,C [N-1]。,用窗宽为5的移动平均滤波器对曲线c滤波可以得到平滑的曲线f。计算公式如下:

如下图,平滑曲线的值是在小窗口上平均左侧噪声曲线的值。下图显示了左侧噪声曲线的示例,使用右侧大小为5移动平均滤波器进行平滑处理。

在C ++版本中,我们定义了一个名为smooth的函数,它计算平滑的移动平均轨迹。

 
  1. /**

  2. * @brief 平滑运动轨迹

  3. *

  4. * @param trajectory 运动轨迹

  5. * @param radius 窗格大小

  6. * @return vector<Trajectory>

  7. */

  8. vector<Trajectory> smooth(vector<Trajectory> &trajectory, int radius)

  9. {

  10. //平滑后的运动轨迹

  11. vector<Trajectory> smoothed_trajectory;

  12. //移动滑动窗格

  13. for (size_t i = 0; i < trajectory.size(); i++)

  14. {

  15. double sum_x = 0;

  16. double sum_y = 0;

  17. double sum_a = 0;

  18. int count = 0;

  19.  
  20. for (int j = -radius; j <= radius; j++)

  21. {

  22. if (i + j >= 0 && i + j < trajectory.size())

  23. {

  24. sum_x += trajectory[i + j].x;

  25. sum_y += trajectory[i + j].y;

  26. sum_a += trajectory[i + j].a;

  27.  
  28. count++;

  29. }

  30. }

  31.  
  32. double avg_a = sum_a / count;

  33. double avg_x = sum_x / count;

  34. double avg_y = sum_y / count;

  35.  
  36. smoothed_trajectory.push_back(Trajectory(avg_x, avg_y, avg_a));

  37. }

  38.  
  39. return smoothed_trajectory;

  40. }

2.2.3 平滑变化计算

到目前为止,我们已经获得了平滑的轨迹。在此步骤中,我们将使用平滑轨迹来获得平滑变换,这些变换可应用于视频帧以使其稳定。这是通过找到平滑轨迹和原始轨迹之间的差异并将该差异添加原始变换矩阵来完成的。C++代码:

 
  1. //平滑后的运动信息结构体

  2. vector<TransformParam> transforms_smooth;

  3.  
  4. //原始运动信息结构体

  5. for (size_t i = 0; i < transforms.size(); i++)

  6. {

  7. // Calculate difference in smoothed_trajectory and trajectory 计算平滑后的轨迹和原始轨迹差异

  8. double diff_x = smoothed_trajectory[i].x - trajectory[i].x;

  9. double diff_y = smoothed_trajectory[i].y - trajectory[i].y;

  10. double diff_a = smoothed_trajectory[i].a - trajectory[i].a;

  11.  
  12. // Calculate newer transformation array 计算平滑后的运动信息结构体数据

  13. double dx = transforms[i].dx + diff_x;

  14. double dy = transforms[i].dy + diff_y;

  15. double da = transforms[i].da + diff_a;

  16.  
  17. transforms_smooth.push_back(TransformParam(dx, dy, da));

  18. }

2.3 将平滑后的变化矩阵应用于帧

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

如果我们将运动指定为x,y,θ ,则相应的变换矩阵由下式给出:

当我们稳定视频时,我们可能会看到一些黑色边界。这是预期的,因为要稳定视频,视频帧原图像可能不得不缩小(不是图的尺寸缩小,图像尺寸不变。有两种情况,一种实际先缩小图像,然后从中截取原图大小的区域,缺少图像区域用黑色填充,起到图像增大作用;另外将原图扩大,然后截取原图尺寸相等大小区域,起到图像缩小作用)。我们可以通过以其视频中点为中心缩放图像(例如4%)来缓解该问题。下面的函数fixBorder显示了实现。我们使用getRotationMatrix2D,因为它可以在不移动图像中心的情况下缩放和旋转图像。我们需要做的就是调用此函数,旋转0和缩放1.04(将原图扩大为1.04倍,然后截取原图尺寸相等大小区域)。C++代码如下:

 
  1. /**

  2. * @brief

  3. *

  4. * @param frame_stabilized

  5. */

  6. void fixBorder(Mat &frame_stabilized)

  7. {

  8. //将原图扩大为1.04倍,然后截取原图尺寸相等大小区域

  9. Mat T = getRotationMatrix2D(Point2f(frame_stabilized.cols / 2, frame_stabilized.rows / 2), 0, 1.04);

  10. //仿射变换

  11. warpAffine(frame_stabilized, frame_stabilized, T, frame_stabilized.size());

  12. }

3 结果和代码

3.1总结

优点:

  • 该方法对低频运动(较慢的振动)提供了良好的稳定性。
  • 该方法具有低内存消耗,因此非常适用于嵌入式设备(如Raspberry Pi)。
  • 此方法可以很好地防止视频中的缩放(缩放)抖动。

缺点:

  • 该方法对高频扰动的影响很小。
  • 速度过慢。
  • 如果运动模糊,则功能跟踪将失败,结果将不是最佳。
  • 滚动快门失真也不适合这种方法。

3.2 代码

代码地址:

https://download.csdn.net/download/luohenyj/11007133

https://github.com/luohenyueji/OpenCV-Practical-Exercise

如果没有积分(系统自动设定资源分数)看看参考链接。我搬运过来的,大修改没有。

代码提供了C++和Python版本,代码都有详细的注释。

但是需要注意的是提供的示例detect.mp4在1300帧后有丢帧现象,有些OpenCV版本会出问题,所以设定n_frames=1300只读取前1300帧,运行其他视频需要注意n_frames 的手动设定。

C++:

 
  1. // video_stabilization.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。

  2. //

  3.  
  4. #include "pch.h"

  5. #include <opencv2/opencv.hpp>

  6. #include <iostream>

  7. #include <cassert>

  8. #include <cmath>

  9. #include <fstream>

  10.  
  11. using namespace std;

  12. using namespace cv;

  13.  
  14. // In frames. The larger the more stable the video, but less reactive to sudden panning 移动平均滑动窗口大小

  15. const int SMOOTHING_RADIUS = 50;

  16.  
  17. /**

  18. * @brief 运动信息结构体

  19. *

  20. */

  21. struct TransformParam

  22. {

  23. TransformParam() {}

  24. //x轴信息,y轴信息,角度信息

  25. TransformParam(double _dx, double _dy, double _da)

  26. {

  27. dx = _dx;

  28. dy = _dy;

  29. da = _da;

  30. }

  31.  
  32. double dx;

  33. double dy;

  34. // angle

  35. double da;

  36.  
  37. void getTransform(Mat &T)

  38. {

  39. // Reconstruct transformation matrix accordingly to new values 重建变换矩阵

  40. T.at<double>(0, 0) = cos(da);

  41. T.at<double>(0, 1) = -sin(da);

  42. T.at<double>(1, 0) = sin(da);

  43. T.at<double>(1, 1) = cos(da);

  44.  
  45. T.at<double>(0, 2) = dx;

  46. T.at<double>(1, 2) = dy;

  47. }

  48. };

  49.  
  50. /**

  51. * @brief 轨迹结构体

  52. *

  53. */

  54. struct Trajectory

  55. {

  56. Trajectory() {}

  57. Trajectory(double _x, double _y, double _a)

  58. {

  59. x = _x;

  60. y = _y;

  61. a = _a;

  62. }

  63.  
  64. double x;

  65. double y;

  66. // angle

  67. double a;

  68. };

  69.  
  70. /**

  71. * @brief 轨迹累积

  72. *

  73. * @param transforms 运动信息结构体

  74. * @return vector<Trajectory> 轨迹结构体

  75. */

  76. vector<Trajectory> cumsum(vector<TransformParam> &transforms)

  77. {

  78. // trajectory at all frames 所有帧的运动轨迹

  79. vector<Trajectory> trajectory;

  80. // Accumulated frame to frame transform 累加计算x,y以及a(角度)

  81. double a = 0;

  82. double x = 0;

  83. double y = 0;

  84.  
  85. //累加

  86. for (size_t i = 0; i < transforms.size(); i++)

  87. {

  88. x += transforms[i].dx;

  89. y += transforms[i].dy;

  90. a += transforms[i].da;

  91.  
  92. trajectory.push_back(Trajectory(x, y, a));

  93. }

  94.  
  95. return trajectory;

  96. }

  97.  
  98. /**

  99. * @brief 平滑运动轨迹

  100. *

  101. * @param trajectory 运动轨迹

  102. * @param radius 窗格大小

  103. * @return vector<Trajectory>

  104. */

  105. vector<Trajectory> smooth(vector<Trajectory> &trajectory, int radius)

  106. {

  107. //平滑后的运动轨迹

  108. vector<Trajectory> smoothed_trajectory;

  109. //移动滑动窗格

  110. for (size_t i = 0; i < trajectory.size(); i++)

  111. {

  112. double sum_x = 0;

  113. double sum_y = 0;

  114. double sum_a = 0;

  115. int count = 0;

  116.  
  117. for (int j = -radius; j <= radius; j++)

  118. {

  119. if (i + j >= 0 && i + j < trajectory.size())

  120. {

  121. sum_x += trajectory[i + j].x;

  122. sum_y += trajectory[i + j].y;

  123. sum_a += trajectory[i + j].a;

  124.  
  125. count++;

  126. }

  127. }

  128.  
  129. double avg_a = sum_a / count;

  130. double avg_x = sum_x / count;

  131. double avg_y = sum_y / count;

  132.  
  133. smoothed_trajectory.push_back(Trajectory(avg_x, avg_y, avg_a));

  134. }

  135.  
  136. return smoothed_trajectory;

  137. }

  138.  
  139. /**

  140. * @brief

  141. *

  142. * @param frame_stabilized

  143. */

  144. void fixBorder(Mat &frame_stabilized)

  145. {

  146. //将原图扩大为1.04倍,然后截取原图尺寸相等大小区域

  147. Mat T = getRotationMatrix2D(Point2f(frame_stabilized.cols / 2, frame_stabilized.rows / 2), 0, 1.04);

  148. //仿射变换

  149. warpAffine(frame_stabilized, frame_stabilized, T, frame_stabilized.size());

  150. }

  151.  
  152. int main(int argc, char **argv)

  153. {

  154. // Read input video 读取视频

  155. VideoCapture cap("./video/detect.mp4");

  156.  
  157. // Get frame count 读取视频总帧数

  158. int n_frames = int(cap.get(CAP_PROP_FRAME_COUNT));

  159. // Our test video may be wrong to read the frame after frame 1300

  160. n_frames = 1300;

  161.  
  162. // Get width and height of video stream 获取视频图像宽高

  163. int w = int(cap.get(CAP_PROP_FRAME_WIDTH));

  164. int h = int(cap.get(CAP_PROP_FRAME_HEIGHT));

  165.  
  166. // Get frames per second (fps) 获取视频每秒帧数

  167. double fps = cap.get(CV_CAP_PROP_FPS);

  168.  
  169. // Set up output video 设置输出视频

  170. VideoWriter out("video_out.avi", CV_FOURCC('M', 'J', 'P', 'G'), fps, Size(2 * w, h));

  171.  
  172. // Define variable for storing frames 定义存储帧的相关变量

  173. //当前帧RGB图像和灰度图

  174. Mat curr, curr_gray;

  175. //前一帧RGB图像和灰度图

  176. Mat prev, prev_gray;

  177.  
  178. // Read first frame 获得视频一张图象

  179. cap >> prev;

  180.  
  181. // Convert frame to grayscale 转换为灰度图

  182. cvtColor(prev, prev_gray, COLOR_BGR2GRAY);

  183.  
  184. // Pre-define transformation-store array 仿射变化参数结构体

  185. vector<TransformParam> transforms;

  186.  
  187. //previous transformation matrix 上一张图像的仿射矩阵

  188. Mat last_T;

  189. //从第二帧开始循环遍历视频所有帧

  190. for (int i = 1; i < n_frames; i++)

  191. {

  192. // Vector from previous and current feature points 前一帧角点vector,当前帧角点vector

  193. vector<Point2f> prev_pts, curr_pts;

  194.  
  195. // Detect features in previous frame 获取前一帧的角点

  196. //前一帧灰度图,前一帧角点vector, 最大角点数,检测到的角点的质量等级,两个角点之间的最小距离

  197. goodFeaturesToTrack(prev_gray, prev_pts, 200, 0.01, 30);

  198.  
  199. // Read next frame 读取当前帧图像

  200. bool success = cap.read(curr);

  201. if (!success)

  202. {

  203. break;

  204. }

  205.  
  206. // Convert to grayscale 将当前帧图像转换为灰度图

  207. cvtColor(curr, curr_gray, COLOR_BGR2GRAY);

  208.  
  209. // Calculate optical flow (i.e. track feature points) 光流法追寻特征点

  210. //输出状态矢量(元素是无符号char类型,uchar),如果在当前帧发现前一帧角点特征则置为1,否则,为0

  211. vector<uchar> status;

  212. //输出误差矢量

  213. vector<float> err;

  214. //光流跟踪

  215. //前一帧灰度图像,当前帧灰度图像,前一帧角点,当前帧角点,状态量,误差量

  216. calcOpticalFlowPyrLK(prev_gray, curr_gray, prev_pts, curr_pts, status, err);

  217.  
  218. // Filter only valid points 获取光流跟踪下有效的角点

  219. //遍历角点

  220. auto prev_it = prev_pts.begin();

  221. auto curr_it = curr_pts.begin();

  222. for (size_t k = 0; k < status.size(); k++)

  223. {

  224. if (status[k])

  225. {

  226. prev_it++;

  227. curr_it++;

  228. }

  229. //删除无效角点

  230. else

  231. {

  232. prev_it = prev_pts.erase(prev_it);

  233. curr_it = curr_pts.erase(curr_it);

  234. }

  235. }

  236.  
  237. // Find transformation matrix 获得变换矩阵

  238. //false表示带几何约束的仿射变换,true则是全仿射变化,T为变换矩阵

  239. Mat T = estimateRigidTransform(prev_pts, curr_pts, false);

  240.  
  241. // In rare cases no transform is found.

  242. // We'll just use the last known good transform.

  243. //极少数情况会找不到变换矩阵,取上一个变换为当前变化矩阵

  244. //当然第一次检测就没找到仿射矩阵,算法会出问题,不过概率很低

  245. if (T.data == NULL)

  246. {

  247. last_T.copyTo(T);

  248. }

  249. T.copyTo(last_T);

  250.  
  251. // Extract traslation 提取仿射变化结果

  252. double dx = T.at<double>(0, 2);

  253. double dy = T.at<double>(1, 2);

  254.  
  255. // Extract rotation angle 提取角度

  256. double da = atan2(T.at<double>(1, 0), T.at<double>(0, 0));

  257.  
  258. // Store transformation 存储仿射变化矩阵

  259. transforms.push_back(TransformParam(dx, dy, da));

  260.  
  261. // Move to next frame 进行下一次检测准测

  262. curr_gray.copyTo(prev_gray);

  263.  
  264. cout << "Frame: " << i << "/" << n_frames << " - Tracked points : " << prev_pts.size() << endl;

  265. }

  266.  
  267. // Compute trajectory using cumulative sum of transformations 获取累加轨迹

  268. vector<Trajectory> trajectory = cumsum(transforms);

  269.  
  270. // Smooth trajectory using moving average filter 获取平滑后的轨迹

  271. vector<Trajectory> smoothed_trajectory = smooth(trajectory, SMOOTHING_RADIUS);

  272.  
  273. //平滑后的运动信息结构体

  274. vector<TransformParam> transforms_smooth;

  275.  
  276. //原始运动信息结构体

  277. for (size_t i = 0; i < transforms.size(); i++)

  278. {

  279. // Calculate difference in smoothed_trajectory and trajectory 计算平滑后的轨迹和原始轨迹差异

  280. double diff_x = smoothed_trajectory[i].x - trajectory[i].x;

  281. double diff_y = smoothed_trajectory[i].y - trajectory[i].y;

  282. double diff_a = smoothed_trajectory[i].a - trajectory[i].a;

  283.  
  284. // Calculate newer transformation array 计算平滑后的运动信息结构体数据

  285. double dx = transforms[i].dx + diff_x;

  286. double dy = transforms[i].dy + diff_y;

  287. double da = transforms[i].da + diff_a;

  288.  
  289. transforms_smooth.push_back(TransformParam(dx, dy, da));

  290. }

  291.  
  292. //定位当前帧为第1帧

  293. cap.set(CV_CAP_PROP_POS_FRAMES, 0);

  294. //平滑后的变化矩阵

  295. Mat T(2, 3, CV_64F);

  296. Mat frame, frame_stabilized, frame_out;

  297.  
  298. //对所有帧进行变化得到稳像结果

  299. //跳过第一帧

  300. cap.read(frame);

  301. for (int i = 0; i < n_frames - 1; i++)

  302. {

  303. bool success = cap.read(frame);

  304. if (!success)

  305. {

  306. break;

  307. }

  308. // Extract transform from translation and rotation angle. 提取平滑后的仿射变化矩阵

  309. transforms_smooth[i].getTransform(T);

  310.  
  311. // Apply affine wrapping to the given frame 应用仿射变化

  312. warpAffine(frame, frame_stabilized, T, frame.size());

  313.  
  314. // Scale image to remove black border artifact 去除黑边

  315. fixBorder(frame_stabilized);

  316.  
  317. // Now draw the original and stablised side by side for coolness 将原图和变化后的图横向排列输出到视频

  318. hconcat(frame, frame_stabilized, frame_out);

  319.  
  320. // If the image is too big, resize it.

  321. if (frame_out.cols > 1920)

  322. {

  323. resize(frame_out, frame_out, Size(frame_out.cols / 2, frame_out.rows / 2));

  324. }

  325.  
  326. //imshow("Before and After", frame_out);

  327. out.write(frame_out);

  328. cout << "out frame:" << i << endl;

  329. //waitKey(10);

  330. }

  331.  
  332. // Release video

  333. cap.release();

  334. out.release();

  335. // Close windows

  336. destroyAllWindows();

  337.  
  338. return 0;

  339. }

python:

 
  1. # Import numpy and OpenCV

  2. import numpy as np

  3. import cv2

  4.  
  5.  
  6. def movingAverage(curve, radius):

  7. window_size = 2 * radius + 1

  8. # Define the filter

  9. f = np.ones(window_size)/window_size

  10. # Add padding to the boundaries

  11. curve_pad = np.lib.pad(curve, (radius, radius), 'edge')

  12. # Apply convolution

  13. curve_smoothed = np.convolve(curve_pad, f, mode='same')

  14. # Remove padding

  15. curve_smoothed = curve_smoothed[radius:-radius]

  16. # return smoothed curve

  17. return curve_smoothed

  18.  
  19.  
  20. def smooth(trajectory):

  21. smoothed_trajectory = np.copy(trajectory)

  22. # Filter the x, y and angle curves

  23. for i in range(3):

  24. smoothed_trajectory[:, i] = movingAverage(

  25. trajectory[:, i], radius=SMOOTHING_RADIUS)

  26.  
  27. return smoothed_trajectory

  28.  
  29.  
  30. def fixBorder(frame):

  31. s = frame.shape

  32. # Scale the image 4% without moving the center

  33. T = cv2.getRotationMatrix2D((s[1]/2, s[0]/2), 0, 1.04)

  34. frame = cv2.warpAffine(frame, T, (s[1], s[0]))

  35. return frame

  36.  
  37.  
  38. # The larger the more stable the video, but less reactive to sudden panning

  39. SMOOTHING_RADIUS = 50

  40.  
  41. # Read input video

  42. cap = cv2.VideoCapture('video/detect.mp4')

  43.  
  44. # Get frame count

  45. n_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

  46.  
  47. # Our test video may be wrong to read the frame after frame 1300

  48. n_frames = 1300

  49.  
  50. # Get width and height of video stream

  51. w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))

  52. h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

  53.  
  54. # Get frames per second (fps)

  55. fps = cap.get(cv2.CAP_PROP_FPS)

  56.  
  57. # Define the codec for output video

  58. fourcc = cv2.VideoWriter_fourcc(*'MJPG')

  59.  
  60. # Set up output video

  61. out = cv2.VideoWriter('video_out.avi', fourcc, fps, (2 * w, h))

  62.  
  63. # Read first frame

  64. _, prev = cap.read()

  65.  
  66. # Convert frame to grayscale

  67. prev_gray = cv2.cvtColor(prev, cv2.COLOR_BGR2GRAY)

  68.  
  69. # Pre-define transformation-store array

  70. transforms = np.zeros((n_frames-1, 3), np.float32)

  71.  
  72. for i in range(n_frames-2):

  73. # Detect feature points in previous frame

  74. prev_pts = cv2.goodFeaturesToTrack(prev_gray,

  75. maxCorners=200,

  76. qualityLevel=0.01,

  77. minDistance=30,

  78. blockSize=3)

  79.  
  80. # Read next frame

  81. success, curr = cap.read()

  82. if not success:

  83. break

  84.  
  85. # Convert to grayscale

  86. curr_gray = cv2.cvtColor(curr, cv2.COLOR_BGR2GRAY)

  87.  
  88. # Calculate optical flow (i.e. track feature points)

  89. curr_pts, status, err = cv2.calcOpticalFlowPyrLK(

  90. prev_gray, curr_gray, prev_pts, None)

  91.  
  92. # Sanity check

  93. assert prev_pts.shape == curr_pts.shape

  94.  
  95. # Filter only valid points

  96. idx = np.where(status == 1)[0]

  97. prev_pts = prev_pts[idx]

  98. curr_pts = curr_pts[idx]

  99.  
  100. # Find transformation matrix

  101. # will only work with OpenCV-3 or less

  102. m = cv2.estimateRigidTransform(prev_pts, curr_pts, fullAffine=False)

  103.  
  104. # Extract traslation

  105. dx = m[0, 2]

  106. dy = m[1, 2]

  107.  
  108. # Extract rotation angle

  109. da = np.arctan2(m[1, 0], m[0, 0])

  110.  
  111. # Store transformation

  112. transforms[i] = [dx, dy, da]

  113.  
  114. # Move to next frame

  115. prev_gray = curr_gray

  116.  
  117. print("Frame: " + str(i) + "/" + str(n_frames) +

  118. " - Tracked points : " + str(len(prev_pts)))

  119.  
  120. # Compute trajectory using cumulative sum of transformations

  121. trajectory = np.cumsum(transforms, axis=0)

  122.  
  123. # Create variable to store smoothed trajectory

  124. smoothed_trajectory = smooth(trajectory)

  125.  
  126. # Calculate difference in smoothed_trajectory and trajectory

  127. difference = smoothed_trajectory - trajectory

  128.  
  129. # Calculate newer transformation array

  130. transforms_smooth = transforms + difference

  131.  
  132. # Reset stream to first frame

  133. cap.set(cv2.CAP_PROP_POS_FRAMES, 0)

  134.  
  135. # Write n_frames-1 transformed frames

  136. for i in range(n_frames-2):

  137. # Read next frame

  138. success, frame = cap.read()

  139. if not success:

  140. break

  141.  
  142. # Extract transformations from the new transformation array

  143. dx = transforms_smooth[i, 0]

  144. dy = transforms_smooth[i, 1]

  145. da = transforms_smooth[i, 2]

  146.  
  147. # Reconstruct transformation matrix accordingly to new values

  148. m = np.zeros((2, 3), np.float32)

  149. m[0, 0] = np.cos(da)

  150. m[0, 1] = -np.sin(da)

  151. m[1, 0] = np.sin(da)

  152. m[1, 1] = np.cos(da)

  153. m[0, 2] = dx

  154. m[1, 2] = dy

  155.  
  156. # Apply affine wrapping to the given frame

  157. frame_stabilized = cv2.warpAffine(frame, m, (w, h))

  158.  
  159. # Fix border artifacts

  160. frame_stabilized = fixBorder(frame_stabilized)

  161.  
  162. # Write the frame to the file

  163. frame_out = cv2.hconcat([frame, frame_stabilized])

  164.  
  165. # If the image is too big, resize it.

  166. if(frame_out.shape[1] > 1920):

  167. frame_out = cv2.resize(

  168. frame_out, (frame_out.shape[1]/2, frame_out.shape[0]/2))

  169.  
  170. #cv2.imshow("Before and After", frame_out)

  171. # cv2.waitKey(10)

  172. out.write(frame_out)

  173.  
  174. # Release video

  175. cap.release()

  176. out.release()

  177. # Close windows

  178. cv2.destroyAllWindows()

参考

https://www.learnopencv.com/video-stabilization-using-point-feature-matching-in-opencv/

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页