JIYIK CN >

In Python, numpy module is used to manipulate arrays. There are many modules available in Python that allow us to read and store images.

An image can be thought of as an array of different pixels stored at specific locations with corresponding color codes. Hence, we may come across a situation where we need to convert and save an array as an image.

In this tutorial, we will discuss how to save a numpy array as an image.

Use Image.fromarray()the function to save a numpy array as an image

fromarray()The function is used to create an image memory from the object of the exported array. We can then save the image memory to our desired location by providing the desired path and file name.

For example,

import numpy as np
from PIL import Image

array = np.arange(0, 737280, 1, np.uint8)
array = np.reshape(array, (1024, 720))

im = Image.fromarray(array)
im.save("filename.jpeg")

We first create an array to store the RGB color codes and then export it. We can specify the desired format of the image in the file name. It can be jpeg, pngor any other commonly used image format. This is common to all the methods discussed below.

Use imageio.imwrite()the function to save a numpy array as an image

Earlier, the scipy module had imsave()a function to save a numpy array as an image. However, in recent versions, it has been deprecated and the function image.io()in has been recommended imwrite()for this task and has gained popularity due to its simplicity.

The following code shows how to use this function.

import imageio
import numpy as np

array = np.arange(0, 737280, 1, np.uint8)
array = np.reshape(array, (1024, 720))

imageio.imwrite("filename.jpeg", array)

Use matplotlib.pyplot.imsave()the function to save a NumPy array as an image

The matplotlib module has a variety of functions for manipulating images.

imsave()The function can save the array as an image file.

For example,

import matplotlib.pyplot as plt
import numpy as np

array = np.arange(0, 737280, 1, np.uint8)
array = np.reshape(array, (1024, 720))

plt.imsave("filename.jpeg", array)

Use cv2.imwrite()the function to save a numpy array as an image

The OpenCV module is commonly used for image processing in Python. imwrite()The function in this module can export a numpy array to an image file.

For example,

import cv2
import numpy as np

array = np.arange(0, 737280, 1, np.uint8)
array = np.reshape(array, (1024, 720))

cv2.imwrite("filename.jpeg", array)