CODEWITHSUNDEEP
pythonpython-3.xnumpy
Cole H
Cole H

Reputation: 3

How to Convert a 2D List of Tuples to a 3D Numpy Array Quickly?

I have a stream of images coming in from a source which is, unfortunately a list of list of tuples of RGB values. I want to perform real-time processing on the image, but that code expects a Numpy array of shape (X,Y,3) where X and Y are the image height and width.

X = [[(R, G, B)...]]
img_arr = np.array([*X])

The above works fine but takes nearly a quarter of a second with my images which is obviously too slow. Interestingly, I also need to go back the other direction after the processing is done, and that code (which seems to work) is not so slow:

imgout = map(tuple, flipped_image)

Some relevant other questions:

why is converting a long 2D list to numpy array so slow?

Convert List of List of Tuples Into 2d Numpy Array

Upvotes: 0

Views: 663

Answers (1)

user17242583
user17242583

Reputation:

To answer the title of your question, numpy automatically lists and tuples to numpy arrays, so you can just use np.array(X), which will be about as fast as you can get:

img_arr = np.array(X)

A simple list comprehension will convert it back to the list-list-tuple form:

imgout = [[tuple(Z) for Z in Y] for Y in img_arr]

Code to generate a sample 10x10 X array:

X = [[tuple(Z) for Z in Y] for Y in np.random.randint(0,255,(10,10,3))]

Upvotes: 1

Related Questions