Channel ordering [B, H, W, C] or [B, W,H,C]

vision
1

Hi Community,

It is just an informative question that may be very basic to most of you. I was reading about channel ordering. Pytorch prefers the first channel order like this [B, C, H, W] and TensorFlow support channel last order [B, H, W, C]. So I am wondering to know does change the channel order affect performance?

What does PyTorch prefer first or last?
How can we visualize these images?

Many thanks

2

PyTorch uses channels-first by default and allows you to transform the input as well as model parameters to channels-last as described here, which could be beneficial for mixed-precision training using TensorCores.

You can visualize the images using any library capable of it (e.g. PIL or matplotib).

1 Like
3

Thank you for always being available and quickly reply to every query.

2 Likes