Vectorization of for loop

2 次查看(过去 30 天)
I have this for loop with conditionals inside it.
for i=1:m
for j=1:n
if (flipped_im(i,j,1)>flipped_im(i,j,2)) || (flipped_im(i,j,1)>flipped_im(i,j,3))
mod_im(i,j,:)=255;
else
mod_im(i,j,:)=flipped_im(i,j,:);
end
end
end
How I can turn this loop into vectorized code?
  2 个评论
Stephen23
Stephen23 2019-3-14
"How I can turn this loop into vectorized code?"
Why do you need to do that?
Yaser Alghawi
Yaser Alghawi 2019-3-14
It is an assignment. To optimize the speed of the program.

请先登录,再进行评论。

采纳的回答

Stephen23
Stephen23 2019-3-14
编辑:Stephen23 2019-3-14
In three lines, no loop:
>> new_im = flipped_im;
>> idx = new_im(:,:,1)>new_im(:,:,2) | new_im(:,:,1)>new_im(:,:,3);
>> new_im(idx(:,:,[1,1,1])) = 255;
and checking:
>> isequal(mod_im,new_im)
ans = 1
For MATLAB versions R2016b and later you can simplify the index code:
idx = any(new_im(:,:,1)>new_im(:,:,2:3),3);
  1 个评论
Jan
Jan 2019-3-14
Or in < R2016b:
idx = any(new_im(:,:, [1, 1]) > new_im(:,:, 2:3), 3);

请先登录,再进行评论。

更多回答(1 个)

Yaser Alghawi
Yaser Alghawi 2019-3-14
Thank you Chris and Jan. It did the trick.

类别

Help CenterFile Exchange 中查找有关 Loops and Conditional Statements 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by