基于CNN与Streamlit的疟疾检测模型预测时输入形状不兼容报错求助

错误原因 #

你的CNN模型训练时定义的输入尺寸是(128, 128, 3)，但当前Streamlit代码里把上传的图片统一缩放到了224x224，导致输入形状(1,224,224,3)和模型期望的不匹配，触发了ValueError。

快速修复方案 #

直接修改Streamlit代码里的图片尺寸参数，和模型训练时保持一致：

找到代码里的这一行：

IMG_SIZE = 224

改成：

IMG_SIZE = 128

修改后，resize_rescale函数会把图片缩放到128x128，和模型的输入要求匹配，就能正常预测了。

额外说明 #

如果一定要用224x224的输入尺寸，需要重新训练模型，将模型的输入层定义为(224,224,3)，同时重新处理训练数据集的图片尺寸。但显然直接修改UI里的缩放尺寸是最快捷的解决方式。

内容的提问来源于stack exchange，提问作者soeyu yjj