matlab视频去抖代码

以下是一个简单的 MATLAB 代码，可以用于视频去抖。

%% 读取视频并去抖
clear;clc;
videoFile = 'test.mp4'; % 视频文件路径
v = VideoReader(videoFile); % 创建视频读取器
frameRate = v.FrameRate; % 获取帧率
N = v.NumberOfFrames; % 获取总帧数

% 初始化变量
filteredFrames = zeros(v.Height, v.Width, 3, N);
position = zeros(N, 2);

% 读取视频并去抖
for i = 1:N
    % 读取当前帧
    frame = readFrame(v);
    
    % 使用高斯滤波器平滑当前帧
    sigma = 3; % 高斯滤波器的标准差
    filteredFrame = imgaussfilt(frame, sigma);
    filteredFrames(:,:,:,i) = filteredFrame;
    
    % 计算当前帧和前一帧的位移
    if i > 1
        position(i,:) = position(i-1,:) + getDisplacement(filteredFrames(:,:,:,i-1), filteredFrame);
    end
end

% 保存去抖后的视频
newVideoFile = 'test_denoised.mp4';
writerObj = VideoWriter(newVideoFile, 'MPEG-4');
writerObj.FrameRate = frameRate;
open(writerObj);
for i = 1:N
    % 获取当前帧的位移
    dx = round(position(i,1));
    dy = round(position(i,2));
    
    % 对当前帧进行位移
    frame = filteredFrames(:,:,:,i);
    frame = circshift(frame, [dy, dx]);
    
    % 写入去抖后的帧
    writeVideo(writerObj, frame);
end
close(writerObj);

%% 获取两帧之间的位移
function [dx, dy] = getDisplacement(frame1, frame2)
% 将 RGB 图像转为灰度图像
if size(frame1, 3) == 3
    frame1 = rgb2gray(frame1);
    frame2 = rgb2gray(frame2);
end

% 计算两张图像的梯度
[Gx1, Gy1] = imgradientxy(frame1);
[Gx2, Gy2] = imgradientxy(frame2);

% 计算两张图像的梯度差
deltaGx = Gx2 - Gx1;
deltaGy = Gy2 - Gy1;

% 计算位移
dx = -mean(deltaGx(:));
dy = -mean(deltaGy(:));
end

这个代码的主要思路是对视频中的每一帧进行高斯滤波器平滑，然后计算每一帧与前一帧之间的位移，最后将每一帧根据位移进行位移矫正，得到去抖后的视频。其中，位移的计算使用了两张图像的梯度差，即梯度流算法。