Skip to content

Instantly share code, notes, and snippets.

@mstfldmr

mstfldmr/Convert channels_first to channels_last

Created April 29, 2019 11:56
Show Gist options
  • Save mstfldmr/a34fa6347cbc445a42ba66d657a15347 to your computer and use it in GitHub Desktop.
Save mstfldmr/a34fa6347cbc445a42ba66d657a15347 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Convert channels_first to channels_last
# MXNet uses channels_first data format while Tensorflow uses channels_last data format.
img.shape
# (3, 224, 224)
x = np.moveaxis(img, 0, 2)
x.shape
# (224, 224, 3)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment