关于tensorRT中遇到的几个问题

本文探讨了在使用TensorRT进行目标检测时遇到的两个主要问题:一是同一程序在不同运行间结果的轻微变化,发现softmax层实现存在缺陷;二是相同代码在不同平台如PC的GTX1080与TX2上的检测精度差异,揭示了平台特异性对结果的影响。

关于tensorRT中遇到的几个问题

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

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

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

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值