将高分辨率图像放入numpy数组的最佳ram高效方式是什么



我目前正在处理GAN中使用的图像,但我还没有找到一个ram-efficient way,可以将30000个图像放入Numpy array中以便于使用。如有任何帮助,我们将不胜感激。

我使用的框架:Tensorflow

首先,让我们假设您有足够的磁盘空间来容纳这30000个图像(非常明显(。

答案的其余部分应该是";使用批次";。你永远不想把30000张图像都保存在你的内存中,因为它的空间效率很低。此外,您必须使用GPU来训练GAN。

因此,请将图像保留在驱动器中。使用批处理,每个小批处理几个图像,将其加载到CPU(RAM(,进行一些预处理,将其移动到GPU,训练,然后继续。

最新更新