解决处理图像时形状不匹配问题

我在用vgg网络训练焊接缺陷模型时遇到随便输入一张图片进行预测时出现以下报错：

output with shape [1, 224, 224] doesn't match the broadcast shape [3, 224, 224]

经检查是输入图像是灰度图像，单通道，而期望图像是三通道的

报错时代码如下：

改正后代码如下：

这是使用了PIL库的convert()方法将图像转换为RGB格式。

问题得到解决。