mstfldmr/Convert channels_first to channels_last

Created April 29, 2019 11:56

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/mstfldmr/a34fa6347cbc445a42ba66d657a15347.js"></script>
Save mstfldmr/a34fa6347cbc445a42ba66d657a15347 to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

mstfldmr created this gist Apr 29, 2019.

Convert channels_first to channels_last

-Original file line number
+Diff line change
@@ -0,0 +1,8 @@
+    # 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)