Store an image in an array using Python

Question

I need to read images (500x300x3) and store them in arrays:

import os
import numpy as np

data = np.empty((number_of_images, 3, 300, 500), dtype="float32")

imgs = os.listdir("./images")
num = len(imgs)
for i in range(num):
    img = Image.open("./images/" + imgs[i])
    arr = np.asarray(img, dtype="float32")
    data[i, :, :, :] = arr

now here error occurs because the size of arr is 500x300x3, but I want to store it as np.empty((number_of_images, 3, 300, 500), dtype="float32").

Any suggestions will be greatly appreciated!

eumiro · Accepted Answer · 2015-12-04 10:33:23Z

1

Transpose the array:

data[i, :, :, :] = arr.transpose(2, 1, 0)

answered Dec 4, 2015 at 10:33

eumiro

214k36 gold badges307 silver badges264 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Store an image in an array using Python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related