1.芯片型号:J5
2.天工开物开发包OpenExplorer版本:J5_OE_v1.1.49b
3.问题定位:板端部署
4.问题具体描述:
您好,我参考《【参考算法】地平线 Swin-T 参考算法-v1.1.0》文章及OE包中horizon_runtime_sample代码,尝试对swint分类模型进行板端部署,horizon_runtime_sample示例代码中是基于ptq量化的bin模型,我按照qat训练的swint进行对应修改,现在板端可以正常推理,但分类结果不正确,与pc端推理结果完全不一致。我的hbm模型完全按照Swin-T 参考算法-v1.1.0步骤获取,现在想请教一下:
1. 量化后上板的hbm模型内部有对输入图片的归一化操作吗?我看到pc端pth模型推理输入图片应该是被归一化到-1–1之间了,板端输入的图片是0–255。
2. hbm模型中最终输出各类别的值没有经过类似softmax的归一化操作吧,这部分若需要,需要自己后处理添加。