チャネルファーストデータをチャネルラストにする - Qiita

はじめに Pytorchを前提に記述されたプログラムをKerasに書き換えている時に、画像データ配列の軸を入れ替える必要があったため、その方法について共有します。 Image Channel Order Channels First : (N, C, H, W) ← PyTorch Channels Last : (N, H, W, C) ← Keras N : 画像の枚数 C : チャネル数（色など） H : 画像のHeight W : 画像のWidth チャネルファースト（Channels First） PyTorchなどにおいて、画像データ配列でモデルを学習する際は、Channels First 形式が一般的です。画像の次元の並び方が **(Channel, Height, Width)**となっています。Channel（Color）の次元が配列の先頭にあることが名前からわかります。