How can I add new dimensions to a Numpy array?
Alternatively to
image = image[..., np.newaxis]in @dbliss' answer, you can also use numpy.expand_dims like
image = np.expand_dims(image, <your desired dimension>)For example (taken from the link above):
x = np.array([1, 2])print(x.shape) # prints (2,)Then
y = np.expand_dims(x, axis=0)yields
array([[1, 2]])and
y.shapegives
(1, 2)
You could just create an array of the correct size up-front and fill it:
frames = np.empty((480, 640, 3, 100))for k in xrange(nframes): frames[:,:,:,k] = cv2.imread('frame_{}.jpg'.format(k))if the frames were individual jpg file that were named in some particular way (in the example, frame_0.jpg, frame_1.jpg, etc).
Just a note, you might consider using a (nframes, 480,640,3) shaped array, instead.