Python: reverse of transferring image into list of pixels?

Question

im = Image.open('segmentace_zkusebni_pi.jpg')
pixel = list(im.getdata())

gets me a list of RGB of every pixel in the image. For my thesis (not programming, math) I need also to reverse this process, after I do some calculations with RGB values of the pixels. So I need to get the list of 3-elemental vectors (variable pixel) transfer into image. Is there any way? Thank you very much

question from:https://stackoverflow.com/questions/65851446/python-reverse-of-transferring-image-into-list-of-pixels

Answer 1

import numpy as np

im = Image.open('segmentace_zkusebni_pi.jpg')
n = np.asarray(im)
shape = n.shape

pixel = list(im.getdata())
# pixel = n.reshape((shape[0]*shape[1], shape[2])).tolist()

# do your work, but make sure the image dimension is kept same
# preferably you should work with numpy array itself
..
..

im2 = Image.fromarray((np.array(pixel, dtype='uint8')).reshape(shape))
im2.show()