matlab实现膨胀卷积的代码

以下是 MATLAB 实现膨胀卷积的代码：

function [output] = dilate_conv(input, kernel)
%DILATE_CONV Perform dilate convolution using MATLAB
%   INPUT:
%       input - input image (2D matrix)
%       kernel - kernel (2D matrix)
%   OUTPUT:
%       output - output image (2D matrix)

% Get size of input image and kernel
[m, n] = size(input);
[kernel_m, kernel_n] = size(kernel);

% Add zero padding to input image
padded_input = padarray(input, [floor(kernel_m/2), floor(kernel_n/2)], 0);

% Initialize output image
output = zeros(m, n);

% Perform dilate convolution
for i = 1:m
    for j = 1:n
        % Get input patch
        input_patch = padded_input(i:i+kernel_m-1, j:j+kernel_n-1);
        % Dilate input patch using kernel
        dilated_patch = input_patch | kernel;
        % Take maximum value of dilated patch and assign it to output
        output(i, j) = max(dilated_patch(:));
    end
end

end

使用方法：

% Load input image and kernel
input = imread('input_image.png');
kernel = [0 1 0; 1 1 1; 0 1 0];

% Convert input image to grayscale
if size(input, 3) == 3
    input = rgb2gray(input);
end

% Perform dilate convolution
output = dilate_conv(input, kernel);

% Display input image, kernel, and output image
subplot(1, 3, 1);
imshow(input);
title('Input Image');
subplot(1, 3, 2);
imshow(kernel);
title('Kernel');
subplot(1, 3, 3);
imshow(output);
title('Output Image');