創建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)