pytorch下训练的resnet18模型转onnx模型后测试onnx模型几乎无精度
收藏回复举报
pytorch下训练的resnet18模型转onnx模型后测试onnx模型几乎无精度
t('forum.solved') 已解决
新人帖
发表于2025-01-15 18:31:39
0 查看

这是resnet18的部分结构

cke_616.png

这是测试的结果 

其中sim是resnet18在pytorch下训练的模型的结果,sim-onnx是转换后的结果 ,并且查看转换后的sim-onnx最小值是0.9以上,无法实现通过设置阈值来判断行为。

cke_33022.png

这是转模型所使用的代码

cke_30262.png

本帖最后由 匿名用户2025/01/16 21:06:03 编辑

我要发帖子