创建Numpy图像阵列

Question

我有（200,1,50,50）形状的图像的numpy数组。这是50张50x50图片的200张图片。

我想翻转图像的一个百分比，并将它们保存在一个全新的阵列中。我可以从原始数组中取出图像并将其翻转，但是我无法弄清楚如何将图像放入一个新数组（需要与原始图像相同的形状（x，1,50,50）。）

如何从（50,50）去翻动的图像在一个新的数组的（0,1,50,50）入口？

Answer 1

如果你想额外维度添加到您阵列，使用内置的功能expand_dims。要翻转图片使用cv2.flip，之后就可以裁剪图像的后处理步骤。

In [1]: import numpy as np 

In [2]: x = np.random.randn(50, 50) 

In [3]: x.shape 
Out[3]: (50, 50) 

In [4]: np.expand_dims(x, axis=0).shape 
Out[4]: (1, 50, 50) 

In [5]: x_ = np.expand_dims(x, axis=0) 

In [6]: x_ = np.expand_dims(x_, axis=0) 

In [7]: x_.shape 
Out[7]: (1, 1, 50, 50)