Help me to optimise or vectorize the code

1 次查看(过去 30 天)
Ank
Ank 2014-8-28
评论: Ank 2014-8-28
I need to optimize this snippet of code where siz is close to 300. tried vectorization with NDGRID but it takes too much memory and matlab throws an error
finalMat = double(zeros(siz,siz,siz));
for i = 1:size(point3d,1)
for j = 1:size(point3d,2)
finalMat(point3d(i,j,1),point3d(i,j,2),point3d(i,j,3)) = max(value(i,j),finalMat(point3d(i,j,1),point3d(i,j,2),point3d(i,j,3)));
end
end
  4 个评论
Joseph Cheng
Joseph Cheng 2014-8-28
编辑:Joseph Cheng 2014-8-28
I'm curious that it takes too much memory (edit: just re-read and realized siz is not apply to size of point3d. What is the size of point3d?) and what the error is. What is the error it gives? is point3d contain non-integer numbers?
point3d = imread('pears.png');
value = rgb2gray(point3d);
siz = 300;
finalMat = double(zeros(siz,siz,siz));
tic
for i = 1:size(point3d,1)
for j = 1:size(point3d,2)
finalMat(point3d(i,j,1),point3d(i,j,2),point3d(i,j,3)) = max(value(i,j),finalMat(point3d(i,j,1),point3d(i,j,2),point3d(i,j,3)));
end
end
toc
I tried the above code as a vague try to experiment with it and it goes quickly and doesn't throw an error.
Ank
Ank 2014-8-28
Currently it gets done in 100 milliseconds but i was looking for it to be faster. Size of point3d is is like 2000x2000.

请先登录,再进行评论。

回答(1 个)

Guillaume
Guillaume 2014-8-28
You could try using accumarray, but I'm not sure it'd be any faster:
rowsubs = point3d(:, :, 1);
colsubs = point3d(:, :, 2);
pagsubs = point3d(:, :, 3);
finalMat = accumarray({rowsubs(:) colsubs(:) pagsubs(:)}, value(:), [siz siz siz], @max);
If you can modify your algorithm to store your point3d coordinates and your values as vectors, you could dispense with all the (:) in accumarray.

类别

Help CenterFile Exchange 中查找有关 Surrogate Optimization 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by