如何将整数与np.array一起保存

Question

我正在做一个ConvNet，需要将一个图像与一个整数（1或0）一起保存，以指示是否已执行操作。 我该怎么做？ 我尝试了这个：

key_pressed = np.array(key_check(self.key)) # key_check() returns 1 or 0
window = np.array(window) # window is opened with cv2

print(window.shape, key_pressed.shape) # (474, 31) (1,)
print(np.concatenate((window, key_pressed)))
      ^^
ValueError: all the input arrays must have same number of dimensions

看到数组的大小不同，用numpy做我想做的事吗？ 关于如何保存此数据的任何想法？

Answer 1

NumPy不会让您简单地将任意大小的东西连接起来：这破坏了数组或矩阵的功能定义。 您正在尝试进行简单的并置，而不是功能串联。 您要做的就是将它们并排显示，对吗？ 这不是矩阵运算。 例如，您是否真的设想对所得的粘合数组进行矩阵乘法？？？

您需要一个简单的容器来容纳两个项目。 我希望一个简单的class能为您完成这项工作。 为结果对编写一个组合的显示方法，并将其用于渲染。

Answer 2

看看scipy.io.savemat和loadmat ：

savemat(your_path, dict(key=key_pressed, window=window))

然后使用loadmat返回带有指定键和所需帧的字典。 甚至还有一个do_compression参数。

编辑：形状不兼容与它无关，因为您不是在创建新数组，而是将键和窗口分别保存在同一文件中。

In [53]: from scipy import io

In [54]: a = np.zeros([3, 4])

In [55]: b = np.ones([5, 17])

In [56]: io.savemat("foo.mat", dict(a=a, b=b))

它就像一个魅力。