如何加速imdb的初始化

Posted 2021-04-16

在使用MATLAB进行深度学习的同时，我在为第一个图像制作imdb时遇到了完整的内存使用。所以我决定用这样的零初始化第一个图像数据：

image_size = [56 56];     
imdb.images.data   = zeros(image_size(1), image_size(2), 1, 30160, 'single');
imdb.images.labels = zeros(1, 30160, 'single');
imdb.images.set    = zeros(1, 30160, 'uint8');

for jj=1:61
    imdb.images.data(:,:,jj,1)=zeros(56,56);
end

我有61个带图像，所以61表示。我的Pc是16 GB RAM和3.4GHZ i7处理器。

有没有办法加快这个初始化？

答案

您可以创建61片56x56零，即三维矩阵，如documentation of zeros中所示，并将它们一次性存储在imdb.images.data中。

imdb.images.data = zeros(56,56,61);