当前位置：首页 > backend >正文

matlab编写的BM3D图像去噪方法

backend 2025/7/12 18:13:22

BM3D（Block-Matching and 3D Filtering）是一种基于块匹配和三维滤波的图像去噪方法，广泛应用于图像处理领域。它通过在图像中寻找相似的块，并将这些块堆叠成三维数组进行滤波处理，从而有效地去除噪声，同时保持图像的细节和结构。

BM3D图像去噪方法的基本原理

块匹配：在图像中寻找相似的块。
三维滤波：将相似的块堆叠成三维数组，并对三维数组进行滤波处理。
硬阈值处理：对滤波后的三维数组进行硬阈值处理，去除小的系数。
反变换：将处理后的三维数组重新变换回二维图像。

BM3D算法的步骤

初始化：对输入图像进行初步去噪。
块匹配：在图像中寻找相似的块。
三维滤波：对相似块进行三维滤波。
硬阈值处理：去除小的系数。
反变换：将处理后的块重新组合成图像。
最终去噪：对初步去噪后的图像进行第二次去噪。

MATLAB代码实现

以下是一个简化的BM3D图像去噪的MATLAB代码实现：

function denoised_image = BM3D_denoising(noisy_image, sigma)% 输入参数：% noisy_image - 带噪声的图像% sigma - 噪声的标准差% 参数设置block_size = 8; % 块的大小search_window = 16; % 搜索窗口大小threshold = 2.5 * sigma; % 硬阈值% 初始化[rows, cols] = size(noisy_image);denoised_image = zeros(rows, cols);% 第一步：初步去噪for i = 1:rows-block_size+1for j = 1:cols-block_size+1% 提取当前块current_block = noisy_image(i:i+block_size-1, j:j+block_size-1);% 寻找相似块similar_blocks = [];for m = max(1, i-search_window):min(rows-block_size, i+search_window)for n = max(1, j-search_window):min(cols-block_size, j+search_window)if m ~= i || n ~= jtemp_block = noisy_image(m:m+block_size-1, n:n+block_size-1);if norm(temp_block(:) - current_block(:)) < thresholdsimilar_blocks = cat(3, similar_blocks, temp_block);endendendend% 三维滤波if ~isempty(similar_blocks)similar_blocks = cat(3, current_block, similar_blocks);[U, S, V] = svd(similar_blocks, 'econ');% 硬阈值处理S = S .* (S > threshold);% 反变换filtered_block = U * S * V';denoised_image(i:i+block_size-1, j:j+block_size-1) = filtered_block(:,:,1);elsedenoised_image(i:i+block_size-1, j:j+block_size-1) = current_block;endendend% 第二步：最终去噪% 对初步去噪后的图像进行第二次去噪denoised_image = denoised_image + noisy_image - imfilter(noisy_image, fspecial('average', block_size));
end

使用示例

% 加载图像
image = imread('cameraman.tif');
image = im2double(image);% 添加高斯噪声
sigma = 0.05;
noisy_image = imnoise(image, 'gaussian', 0, sigma^2);% 使用BM3D去噪
denoised_image = BM3D_denoising(noisy_image, sigma);% 显示结果
figure;
subplot(1, 3, 1);
imshow(image);
title('原始图像');subplot(1, 3, 2);
imshow(noisy_image);
title('带噪声图像');subplot(1, 3, 3);
imshow(denoised_image);
title('去噪后的图像');