DLの入力に合わせるためにテンソルの次元を上げたい
今U-NETを使って物体認識の機械学習に取り組んでいるところです。
組んだモデルが768*768のRGB画像なので取り込むと3階のテンソルになります。
実際に入力変数としてshape=(768, 768, 3)のテンソルを準備しました。
しかしうまくいかないのでモデルを図示してみると入力はshape=[None, 768, 768, 3]
という入力になっていたので3階のテンソルを4階に上げたいのですがネットを探してもうまい方法が見当たりません。
ソースコード(のうち問題の部分)
Python
1#img load 2img=Image.open('image_path' + imgname)] 3 4#img reshape 5img_pixels=np.asarray(img) 6img_reshaped = np.reshape(img_pixels, newshape=(768, 768, 3)) #ここで3階テンソルを作っています 7 8#train 9model.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画像から作ったテンソルが正しく元画像に対応しているのか怪しいところがあります。
あわせてご教授いただけると幸いです。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。