トランスフォームndarrayの形状

Question

私はpython/Numpyを使って、小さな画像をndarrayに保存しています。 ndarrayを32,32,1の形から1,32,32,1に変換しようとすると、私は立ち往生します。どんな助け？ありがとう

Answer 1

numpy配列のサイズを拡張する必要があります。 np.expand_dimsを使用してください。

arr = np.expand_dims(arr, axis=0) 

Answer 2

arr[np.newaxis, :, :, :]代わりに明示的にも明示的に軸を追加し、それを再構築でき軸追加の

Answer 3

動作します：

>>> import numpy as np 
>>> arr = np.ones((32, 32, 1)) # just ones for demonstration purposes 

>>> reshaped = arr.reshape(1, *arr.shape) 
>>> reshaped.shape 
(1, 32, 32, 1)