在 Octave / matlab 代码中加速 for 循环

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：我使用的是 Octave 4.0，类似于 matlab

Answer 1

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

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]);。
2. 直接通过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 以避免在内存中重新分配数组。

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