加速 Octave / matlab 代码中的循环

Question

完成这些循环大约需要 2 分钟，除了更改颜色数量或使用的单元格数量之外，有人知道我可以如何加快速度/更快地创建图像吗？矢量化会有帮助吗？如果有的话，我应该怎么做？基本上它会在每个循环中创建一个动画帧

查看下面的代码

clear all,clf reset,tic,clc,clf

rgb1 = jet(256);
len1 = size(rgb1,1);
RGB1 = permute(rgb1,[3 1 2]);
figure; imshow(RGB1); %bar of colors
len2 = 200;

num2use=numel(num2str(length(rgb1)))+1 %count how many values and add 1. can use R also
lead_zeros=strcat('%0',num2str(num2use),'d'); %creates variable needed

for ii=0:length(rgb1)
  ii
  rgb1_shift=circshift(rgb1,[ii,:]);

  t = linspace(0,4*pi,len2);
  x = t.*cos(t);
  y = t.*sin(t);
  %rgb2 = interp1(1:len1,rgb1,linspace(1,len1,len2));
  rgb2 = interp1(1:len1,rgb1_shift,linspace(1,len1,len2));

  [xg,yg] = meshgrid([-t(end:-1:2) t],[-t(end:-1:2) t]);
  RGB2 = zeros([size(xg) 3]);
  % interpolate for the desired coordinates
  for c = 1:3
      RGB2(:,:,c) = griddata(x,y,rgb2(:,c),xg,yg);
  end
  imwrite(RGB2,strcat('/tmp/img2/',sprintf(lead_zeros, ii),'_spiral','.png')); %will create file without borders and use any resize in repmat

end


fprintf('\nfinally Done-elapsed time -%4.4fsec- or -%4.4fmins- or -%4.4fhours-\n',toc,toc/60,toc/3600);

Ps: 我正在使用类似于 matlab

的 Octave 4.0

Answer 1

这里有一些错误，修复它们可以大大提高速度：

将所有循环自变量移出循环。
即，将 t = linspace(0,4*pi,len2);、x = t.*cos(t);、y = t.*sin(t);、[xg,yg] = meshgrid([-t(end:-1:2) t],[-t(end:-1:2) t]); 和 RGB2 = zeros([size(xg) 3]); 移出循环。
直接用RGB2(:) = cat(3, griddata(x,y,rgb2(:,1),xg,yg), griddata(x,y,rgb2(:,2),xg,yg), griddata(x,y,rgb2(:,3),xg,yg))定义RGB2，避免在内存中重新分配数组。

如果您能解释一下您到底想做什么，我们可能会更有帮助。

加速 Octave / matlab 代码中的循环

Speeding up for loops in Octave / matlab code

matlab

for-loop

vectorization

octave