首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Tensorflow-Onnx-Tensorrt的准确度下降

Tensorflow-Onnx-Tensorrt的准确度下降
EN

Stack Overflow用户
提问于 2020-09-11 01:13:13
回答 1查看 418关注 0票数 0

我有一个tensorflow训练模型,并在tensorflow测试,精度达到95%。

Tensorflow模型被转换为ONNX并转换为TensorRT。TensorRT引擎以16位精度运行.在TensorRT中,准确率下降到75%。即使使用kTF32,准确率仍为75%。测试相同的图像为两个测试和相同的输入大小。

我该从哪里看这个精度下降的地方?唯一的区别是我在TensorRT中进行批处理推理,而不是在Tensorflow中。

对于Tensorflow和TensorRT,使用OpenCV读取图像。

从Tensorflow到ONNX的转换精度会下降吗?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-09-14 06:26:32

现在我发现了这个问题。Tensorflow通过乘以1/255.0对输入图像进行归一化处理。但在张力归一化,它是1-x/255.0。这就是问题所在。现在我也有了同样的准确性。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/63839519

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档