将单张图片输入CNN模型时出现形状不兼容错误的解决方法

错误原因 #

你的CNN模型训练时使用的是3通道RGB图像，期望输入形状为(None, 54, 50, 3)，但你读取的图片是带透明通道的RGBA格式（4通道），导致输入形状变为(None, 54, 50, 4)，和模型要求不兼容。

解决方案 #

在读取图片后，用PIL的convert('RGB')方法将RGBA图像转为3通道RGB图像，丢弃透明通道即可：

from PIL import Image

def test_img(file):
    img = Image.open(file).convert('RGB')  # 转换为3通道RGB图像
    ih = ImageHandler([img]).resize((54,50)).normalize()
    
    ogmodel = models.load_model('../models/best_original.h5')
    
    result = (ogmodel.predict(ih.images) > 0.5).astype("int32")
    
    print(result)

test_img("../bees/test_imgs/beex.png")

补充说明 #

• convert('RGB')会自动处理透明通道，将RGBA格式统一转为模型需要的3通道格式，确保输入形状匹配。
• 若你的ImageHandler类在后续处理中有修改通道数的逻辑，也可以检查其实现，确保最终输出的图像是3通道。

内容的提问来源于stack exchange，提问作者Ahleya Sheikh