画像データの保存形の変換方法

Question

例えば、dim = ３×３のRGB系のカラー画像があるとします。
画像ファイルの画素データが以下の形で保存されている場合

RGB RGB RGB
RGB RGB RGB
RGB RGB RGB

を下記の形のtensorに変換したいですが、一番簡単な方法(例えば、たった一行のPythonの文で)を教えてください。

RRR
RRR
RRR

GGG
GGG
GGG

BBB
BBB
BBB

Answer 1

import numpy as np

img = np.array([
  [[1, 11, 101], [4, 14, 104], [7, 17, 107]], 
  [[2, 12, 102], [5, 15, 105], [8, 18, 108]], 
  [[3, 13, 103], [6, 16, 106], [9, 19, 109]], 
])

print(img.transpose(2, 0, 1))

#
[[[  1   4   7]
  [  2   5   8]
  [  3   6   9]]

 [[ 11  14  17]
  [ 12  15  18]
  [ 13  16  19]]

 [[101 104 107]
  [102 105 108]
  [103 106 109]]]

Comments

[insecticide]

凄い！　理解できていませんが、こんな方法もありますね。

勉強します。

本当にありがとうございました！！