How to avoid this for loops to improve computation time

Question

Gopichandh Danala 2017-2-3

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/323284-how-to-avoid-this-for-loops-to-improve-computation-time

评论： Gopichandh Danala 2017-2-5

In this code, I have a cluster image with 10 classes and i want to extract 10 different images for each level and save as a 10 images Below is the code, I used

tic
numberOfClasses = 10;
segment_label_images = cell(1,numberOfClasses);
pixelCount = zeros(1,numberOfClasses);
[rs, cs] = size(classImage);
% classImage has intensity range from 1-numberOfClasses
for k = 1:numberOfClasses
    for i = 1:rs
        for j = 1:cs
            if classImage(i,j) == k
                segment_label_images{k}(i,j) = 1;
            else
                segment_label_images{k}(i,j) = 0;
            end
        end
    end
pixelCount(k) = sum(segment_label_images{k}(:));
%figure, imshow(segment_label_images{k},[]);
end
toc

Here, I have 3 for loops and I think that is affecting computational time. Elapsed time is 0.089413 seconds.

Any suggestions to avoid for loop to improve comp time.? Thanks, Gopi

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Guillaume 2017-2-4

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/323284-how-to-avoid-this-for-loops-to-improve-computation-time#answer_253293

编辑：Guillaume 2017-2-4

在 MATLAB Online 中打开

First issue: you've predeclared segment_label_images and pixelCount however, you don't predeclare the matrices in each cell of segment_label_images, so these grow at each step of the i and j loops.

Workaround: before the i loop:

for k = 1:numerofClasses
    segment_label_images{k} = zeros(size(classImage));
    for i = 1:rs
        %...

However, there's no point in the i and j loop, and thus actually no need to predeclare the matrices:

for k = 1:numberofClasses
    segment_label_images{k} = classImage == k;
    pixelcount(k) = sum(segment_label_images{k}(:));
    figure;
    imshow(segment_label_images{k});    %no need for []
end