如何加快imdb的初始化

Question

在使用 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 表示。我的电脑是 16 GB RAM 和 3.4GHZ i7 处理器。

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

Answer 1

您可以创建 61 个 56x56 零的切片，即一个 3D 矩阵，如 documentation of zeros 中给出的那样，并将它们一次性存储在 imdb.images.data 中。

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