Pythonでテンソルの次元を上げたくて困っています

Question

DLの入力に合わせるためにテンソルの次元を上げたい

今U-NETを使って物体認識の機械学習に取り組んでいるところです。
組んだモデルが768*768のRGB画像なので取り込むと3階のテンソルになります。
実際に入力変数としてshape=(768, 768, 3)のテンソルを準備しました。

しかしうまくいかないのでモデルを図示してみると入力はshape=[None, 768, 768, 3]
という入力になっていたので3階のテンソルを4階に上げたいのですがネットを探してもうまい方法が見当たりません。

ソースコード（のうち問題の部分）

#img load
img=Image.open('image_path' + imgname)]

#img reshape
img_pixels=np.asarray(img)
img_reshaped = np.reshape(img_pixels, newshape=(768, 768, 3)) #ここで3階テンソルを作っています

#train
model.fit(x=img_reshaped, y=label)

エラーコードは下のようなものが出ます。

ValueError: Error when checking input: expected input_1 to have 4 dimensions, but got array with shape (768, 768, 3)

初心者なもので恥ずかしながらRGB画像から作ったテンソルが正しく元画像に対応しているのか怪しいところがあります。

あわせてご教授いただけると幸いです。

Answer 1

numpy.expand_dims() または x[np.newaxis, :] で次元を増やすことができます。

import numpy as np

x = np.zeros((300, 300, 3))

x1 = np.expand_dims(x, axis=0)
print(x1.shape)  # (1, 300, 300, 3)

x2 = x[np.newaxis, :]
print(x2.shape)  # (1, 300, 300, 3)