关于tensorRT中遇到的几个问题

关于tensorRT中遇到的几个问题

一、运行同一个程序,前后两次的运行结果不一样
例如运行tensorRT-SSD代码,同样的测试图片,检测出来的目标框会有轻微的抖动,但是目标还是能检测出来,不会影响精度。

2018-11-5
找到原因所在了,softmax层的实现有问题。
只要是程序运行出现随机现象,那么一定是有问题的。一个正常的程序运行,如果自己没有写随机代码,那么程序肯定是有问题的。有很大可能是某个变量在定义的时候没有初始化。

二、相同的代码,相同的模型,相同的测试图片,在不同的平台上结果不一样
我在pc上的gtx1080显卡运行tensorRT代码,全部调试通过之后移植到tx2平台上运行,却发现在tx2平台上的检测精度要比pc上低一些

同一份代码,如果保证是三方库是一样的,绝不会出现一样的结果。要么就是针对不同的平台,相同版本的三方库的实现也是不一样的

发布了65 篇原创文章 · 获赞 63 · 访问量 15万+

猜你喜欢

转载自blog.csdn.net/qq_17278169/article/details/83350608
今日推荐