What is the correct way to change image channel ordering between channels first and channels last?

python numpy machine-learning keras theano
Kuysea picture Kuysea · May 7, 2017 · Viewed 41.1k times · Source

I can not for the life of me figure out how to switch the image ordering. images are read in (x,x,3) format, theano requires it to be in (3,x,x) format. I tried changing the order with numpy.array([img[:,:,i] for i in range(3)])

which i guess gets the job done, but it is both ugly and i can't figure out how to reverse it to get the original image back.

Answer

Daniel Möller picture Daniel Möller · May 7, 2017

To reorder data

You can use numpy.rollaxis to roll the axis 3 to position 1 (considering you have the batch size as dimension 0).

np.rollaxis(imagesArray, 3, 1)

But, if you're using keras, you might want to change its configuration or define it per layer. Theano doesn't require anything from you if you're using Keras.

Keras can be configured with channels first or channels last, besides allowing you to define it in every individual layer, so you don't have to change your data.

To configure keras

Find the keras.json file and change it. The file is usually installed in C:\Users\yourusername\.keras or ~/.keras depending on your OS.

Change "image_data_format": "channels_last" to "channels_first" or vice-versa, as you wish.

Usually, working with "channels_last" is less troublesome because of a great amount of other (non convolutional) functions that work only on the last axis.

Defining channel order in layers.

The Keras documentation has all information about parameters for layers, including the data_format parameter.

Related questions

How to implement the Softmax function in Python

From the Udacity's deep learning class, the softmax of y_i is simply the exponential divided by the sum of exponential of the whole Y vector: Where S(y_i) is the softmax function of y_i and e is …

Read More
Convert array of indices to 1-hot encoded numpy array

Let's say I have a 1d numpy array a = array([1,0,3]) I would like to encode this as a 2D one-hot array b = array([[0,1,0,0], [1,0,0,0], [0,0,0,1]]) Is there a quick way to do this? Quicker than just looping over a to set elements …

Read More
gradient descent using python and numpy

def gradient(X_norm,y,theta,alpha,m,n,num_it): temp=np.array(np.zeros_like(theta,float)) for i in range(0,num_it): h=np.dot(X_norm,theta) #temp[j]=theta[j]-(alpha/m)*( np.sum( (…

Read More