保存numpy图片数据集。无需增加大小,并且易于保存和加载数据

时间:2019-07-11 20:10:47

标签: python numpy

我已经将火车测试Val数组保存到pickle文件中。但图片大小为1.5GB,斑点文件为16GB,即大小增加。还有另一种方法可以在不增加大小的情况下保存那些numpy图片数组?

1 个答案:

答案 0 :(得分:0)

使用numpy.save函数(documentation)或numpy.savez_compresion函数(documentation)。

在问问题之前先阅读文档。

示例代码:

import numpy as np 


image = np.random.randint(0, 200, (199818,50,50,3), dtype=np.uint8)
image2 = np.random.randint(0, 200, (1998,50,50,3), dtype=np.uint8)

np.savez_compressed("test.npz", image, img=image2)

img_dkt = np.load("test.npz")

print("First_array:", img_dkt["arr_0"].shape, "equality", np.all(image == img_dkt["arr_0"]))
print("second_array:", img_dkt["img"].shape, "equality", np.all(image2 == img_dkt["img"]))