numpyの中pytorch交換チャネルと問題点

フォーマットpytorchネットワーク入力画像は、（C、H、W）であり、そして形状numpyの画像である（H、W、C）

したがって、チャンネルを変更する一般的な必要性は、形状が形状のnumpyのトーチに変換されます。

ここに方法は次のとおりです。

＃numpyのデータ型がされ
、A = A.transpose（0,1,2）  ＃変更されない：（H、W、C） 
A = A.transpose（2,0,1）   ＃は、（C、H：に変換されます。 ）のw

このトーチは、ネットワークに直接入力することができます。