Как использовать новые функции набора инструментов PDE в обработке изображений (Matlab R2015a)

Я хочу использовать команду line new панели инструментов PDE в качестве Matlab R2015 для восстановления зашумленного изображения с гауссовским шумом.

PDE - это:

(. (( u) / (√ (1+ | u | 2))) + (f 2) / (u 2) = 1 в Ω ( u) / ( n) = 0 в Ω

Где f - шумное изображение, а u - восстановленное изображение.

Я попробовал следующий код:

clear 
close all
clc

img = 'AA.jpg';
mInputImage = double(imread(img));
mInputImage = rgb2gray(mInputImage);
[numRows, numCols] = size(mInputImage);

Var = 0.04;
Mean = 0;
mInputImageNoisy = imnoise((mInputImage(:,:,1)),'gaussian',Mean, Var); 

% reshape the input and noisy images to vectors
mInputImageVector = reshape(mInputImage,numRows*numCols,1);
mInputImageNoisyVector = reshape(mInputImageNoisy,numRows*numCols,1);

Residu1 = norm(mInputImageVector-mInputImageNoisyVector)/norm(mInputImageVector)

RegularisationCoefficient = 0.7*ones((numRows-1)*(numCols-1),1);
mOutputImageVector = mInputImageNoisyVector;
%a = (mInputImageNoisyVector.^2) ./ mOutputImageVector.^3;
f = 1;
rtol = 1e-1;
c = '1./sqrt(1+ux.^2+uy.^2)';

% Create a PDE Model with a single dependent variable
numberOfPDE = 1;
pdem = createpde(numberOfPDE);

g = @squareg;
geometryFromEdges(pdem,g);

% Plot the geometry and display the edge labels for use in the boundary
% condition definition.
figure;
pdegplot(pdem, 'edgeLabels', 'on');
%axis([0 numRows 0 numCols]);
axis([-2 2 -2 2]);
title 'Geometry With Edge Labels Displayed'


b2 = applyBoundaryCondition(pdem,'Edge',[1 2 3 4], 'u', 0);

[p,e,t] = poimesh(g,numRows, numCols);
numCols
pdemesh(p,e,t);
axis equal

for iter = 1: numRows*numRows,
mOutputImageVector(iter) = pdenonlin(pdem,c,...
     (mInputImageNoisyVector(iter).^2) ./ mOutputImageVector(iter).^3,...
              f,'tol',rtol);
SaveImageVector(iter) = mOutputImageVector;
end

mOutputImage = reshape(SaveImageVector,numRows,numRows);

mOutputImage = uint8(mOutputImage);

figure()
imshow(mOutputImage)

0 ответов

Другие вопросы по тегам