94 lines
2.4 KiB
Mathematica
94 lines
2.4 KiB
Mathematica
|
%--------------------------------------------------------------------------
|
|||
|
|
% ENSEEIHT - 2SN MM - Traitement des donn<EFBFBD>es audio-visuelles
|
|||
|
|
% TP6 - Restauration d'images
|
|||
|
|
% exercice_3 : inpainting par rapi<EFBFBD><EFBFBD>age (domaine D connu)
|
|||
|
|
%--------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
clear
|
|||
|
|
close all
|
|||
|
|
clc
|
|||
|
|
|
|||
|
|
% Mise en place de la figure pour affichage :
|
|||
|
|
taille_ecran = get(0,'ScreenSize');
|
|||
|
|
L = taille_ecran(3);
|
|||
|
|
H = taille_ecran(4);
|
|||
|
|
figure('Name','Inpainting par rapiecage',...
|
|||
|
|
'Position',[0.06*L,0.1*H,0.9*L,0.75*H])
|
|||
|
|
|
|||
|
|
% Lecture de l'image :
|
|||
|
|
u0 = double(imread('Images/randonneur.jpg'));
|
|||
|
|
[nb_lignes,nb_colonnes,nb_canaux] = size(u0);
|
|||
|
|
u_max = max(u0(:));
|
|||
|
|
|
|||
|
|
% Affichage de l'image :
|
|||
|
|
subplot(1,2,1)
|
|||
|
|
imagesc(max(0,min(1,u0/u_max)),[0 1])
|
|||
|
|
axis image off
|
|||
|
|
title('Image originale','FontSize',20)
|
|||
|
|
if nb_canaux == 1
|
|||
|
|
colormap gray
|
|||
|
|
end
|
|||
|
|
|
|||
|
|
% Lecture du domaine <EFBFBD> restaurer :
|
|||
|
|
D = imread('Images/masque_randonneur.png') > 0;
|
|||
|
|
|
|||
|
|
% Affichage de l'image r<EFBFBD>sultat :
|
|||
|
|
u_k = u0;
|
|||
|
|
for c = 1:nb_canaux
|
|||
|
|
u_k(:,:,c) = (~D).*u_k(:,:,c);
|
|||
|
|
end
|
|||
|
|
subplot(1,2,2)
|
|||
|
|
imagesc(max(0,min(1,u_k/u_max)),[0 1])
|
|||
|
|
axis image off
|
|||
|
|
title('Image resultat','FontSize',20)
|
|||
|
|
if nb_canaux == 1
|
|||
|
|
colormap gray
|
|||
|
|
end
|
|||
|
|
drawnow nocallbacks
|
|||
|
|
|
|||
|
|
% Lancement du traitement :
|
|||
|
|
fprintf('Tapez un caractere pour lancer le traitement !\n');
|
|||
|
|
pause
|
|||
|
|
|
|||
|
|
% Initialisation de la fronti<EFBFBD>re de D :
|
|||
|
|
delta_D = frontiere(D);
|
|||
|
|
indices_delta_D = find(delta_D > 0);
|
|||
|
|
nb_points_delta_D = length(indices_delta_D);
|
|||
|
|
|
|||
|
|
% Param<EFBFBD>tres :
|
|||
|
|
t = 9; % Voisinage d'un pixel de taille (2t+1) x (2t+1)
|
|||
|
|
T = 50; % Fen<EFBFBD>tre de recherche de taille (2T+1) x (2T+1)
|
|||
|
|
|
|||
|
|
% Tant que la fronti<EFBFBD>re de D n'est pas vide :
|
|||
|
|
while nb_points_delta_D > 0
|
|||
|
|
|
|||
|
|
% Pixel p de la fronti<EFBFBD>re de D tir<EFBFBD> al<EFBFBD>atoirement :
|
|||
|
|
indice_p = indices_delta_D(randi(nb_points_delta_D));
|
|||
|
|
[i_p,j_p] = ind2sub(size(D),indice_p);
|
|||
|
|
|
|||
|
|
% Recherche du pixel q_chapeau :
|
|||
|
|
[existe_q,bornes_V_p,bornes_V_q_chapeau] = d_min(i_p,j_p,u_k,D,t,T);
|
|||
|
|
|
|||
|
|
% S'il existe au moins un pixel q <EFBFBD>ligible :
|
|||
|
|
if existe_q
|
|||
|
|
|
|||
|
|
% Rapi<EFBFBD><EFBFBD>age et mise <EFBFBD> jour de D :
|
|||
|
|
[u_k,D] = rapiecage(bornes_V_p,bornes_V_q_chapeau,u_k,D);
|
|||
|
|
|
|||
|
|
% Mise <EFBFBD> jour de la fronti<EFBFBD>re de D :
|
|||
|
|
delta_D = frontiere(D);
|
|||
|
|
indices_delta_D = find(delta_D > 0);
|
|||
|
|
nb_points_delta_D = length(indices_delta_D);
|
|||
|
|
|
|||
|
|
% Affichage de l'image r<EFBFBD>sultat :
|
|||
|
|
subplot(1,2,2)
|
|||
|
|
imagesc(max(0,min(1,u_k/u_max)),[0 1])
|
|||
|
|
axis image off
|
|||
|
|
title('Image resultat','FontSize',20)
|
|||
|
|
if nb_canaux == 1
|
|||
|
|
colormap gray
|
|||
|
|
end
|
|||
|
|
drawnow nocallbacks
|
|||
|
|
end
|
|||
|
|
end
|