Vectorisation des boucles

Question

Dans MATLAB, le code vectorisé est plus rapide que l'utilisation de boucles for-loops. J'ai essayé de le faire mais je ne comprends pas complètement comment cela fonctionne. J'espérais que quelqu'un pourrait me montrer comment améliorer ces 2 boucles et une boucle avec des indices changeants pour que je puisse tourner la tête. Toute aide serait incroyable.

width= 700; 
height= 600; 
fg= zeros(height, width); 
for i= 1: height 
    for j= 1: width 
     fg(i, j) = 0; 
     while ((match== 0)&& (k<= M)) 
      if (w(i, j, rank_ind(k))>= thresh) 
       if (abs(u_diff(i, j, rank_ind(k)))<= D* sd(i, j, rank_ind(k))) 
        fg(i, j)= 0; 
        match= 1; 
       else 
        fg(i, j)= fr_bw(i, j); 
       end 
      end 
      k= k+ 1; 
     end 
    end 
end 

Remarque w, u_diff, sd, rank_ind et fr_b sont tous les tableaux

Answer 1

Permettez-moi de voir si je comprends bien: Vous voulez copier la valeur de fr_bw dans fg uniquement si aucune valeur correspondante dans u_diff est plus petit que D*sd, et w est au-dessus d'un certain seuil, non?

Dans ce cas, vous pouvez réécrire votre code de la manière suivante:

%# find where u_diff is larger than D*sd 
%# but not where there's any u_diff that is smaller than D*sd 
whereToCopy = any(w(:,:,rank_ind) >= thresh & ... 
    abs(u_diff(:,:,rank_ind)) > D*sd(:,:,rank_ind),3) & ... 
    ~any(w(:,:,rank_ind) >= thresh & ... 
    abs(u_diff(:,:,rank_ind)) <= D*sd(:,:,rank_ind),3); 

%# whereToCopy has 1 wherever we need to copy, and 0 elsewhere 
%# thus, create fg from fr_bw directly by multiplication 
fg = double(fr_bw) .* double(whereToCopy);