Process an Image dataset

To load the images from the image dataset, the simple method is to use load_data() on the image dataset. We are using mnist dataset which is already available in Keras. It will give in return x_train, y_train, x_test, and y_test. The x_train and y_train will be used to train the model and x_test and y_test will be used for testing purposes. We can reshape all the images inside the dataset using reshape() method, and define what type of images should be like ‘float64’ or ‘float32’. 

Python3




from keras.datasets import mnist
 
(X_train, Y_train), (X_test, Y_test) = mnist.load_data()
 
# reshape the image
images = X_train.reshape(-1, 28, 28, 1).astype('float64')
 
print(images.shape)
print(type(images))
print(images.size)


Output:

We see in the above output, 60000 images in mnist have been reshaped into 28 x 28 size and images are of type numpy n-dimensional array.



Image Processing with Keras in Python

In this article, we are doing Image Processing with Keras in Python. Keras API is a deep learning library that provides methods to load, prepare and process images. 

We will cover the following points in this article:

  • Load an image
  • Process an image
  • Convert Image into an array and vice-versa
  • Change the color of the image
  • Process image dataset

Similar Reads

Load the Image

In Keras, load_img() function is used to load image. The image loaded using load_img() method is PIL object. Certain information can be accessed from loaded images like image type which is PIL object, the format is JPEG, size is (6000,4000), mode is RGB, etc. We are using dog images throughout the article....

Resize an Image

...

Convert image into an array

We can perform certain functions on the image like resizing it, changing its color, convert into an array, etc before training any model. To resize the shape of the image, resize() method is invoked on the image. The size in which we want to convert the image should be iterable....

Change the color of the Image

...

Process an Image dataset

There is img_to_array() method to convert images into array, and array_to_img() method to convert image array back to image. In the below example, we are just accessing the 0th index of the image array. We can get information like image array shape, type, etc. When the image array is converted to an image, it again is a PIL object....