检查输入形状没有问题，不太清楚到底是哪里出错

这是整个文件包括了很少量的数据集，以及推理om模型和代码

链接：https://pan.quark.cn/s/8244b2daf49b