TensorFlowによる画像認識：InvalidArgumentErrorを解消したい

TensorFlow初心者です。以下のサイトを参考にさせて頂き、MobileNetV2により画像認識をしたいと思っています。現状、サイトは実装しました。その後、自分の画像を用いて試したところ、以下の箇所で、Errorが出てしまいます。海外サイトで同様の質問が出ていましたが、解消方法はわかりませんでした。おそらく、画像データの変換？がうまくいっていないのかと思います。どなたかお分かりになりましたら、アドバイスをお願い致します。

参考サイト

history = model.fit(train_data.repeat(),
                    epochs=num_epochs,
                    steps_per_epoch = steps_per_epoch,
                    validation_data=val_data.repeat(), 
                    validation_steps=val_steps)

■以下、エラーメッセージ

InvalidArgumentError: input depth must be evenly divisible by filter depth: 4 vs 3
[[node sequential_2/mobilenetv2_1.00_96/Conv1/Conv2D (defined at C:\Users\XXXXX\Anaconda3\lib\site-packages\tensorflow_core\python\framework\ops.py:1751) ]] [Op:__inference_distributed_function_42611]
Function call stack:
distributed_function

■環境
Tensorflow ver.2.0.0
python ver.3.8.1
Jupyter notebook

hoshi-takanori

2020/02/12 21:14

画像のアルファチャンネルっぽいですね。img0 = np.float32(img0)[:,:,:3] は試しましたか？ https://github.com/llSourcell/deep_dream_challenge/issues/5#issuecomment-294718597

51sep

2020/02/13 16:47

ご回答をありがとうございました。おっしゃる通り、画像のアルファチャンネルの問題だったようです。ご提示のサイト、その他のサイトを参考にさせて頂き、以下のプログラムで解消しました。 XXX = tf.convert_to_tensor(XXX[:,:,:3]) どうもありがとうございました。皆様にも参考になりますように。