YOLOX icon indicating copy to clipboard operation
YOLOX copied to clipboard

TensorRT/C++ 进行推理的时候,出来的结果不对

Open EkilGitHub opened this issue 3 years ago • 4 comments

运用trt.py将yolox_tiny.pth转换成model_trt.pth, model_trt.engine 其中model_trt.pth 预测的结果和yolox_tiny.pth一样: image 但是model_trt.engine预测的结果狠奇怪: image

请问是啥原因导致的?

EkilGitHub avatar May 10 '22 11:05 EkilGitHub

Did you check the pre-process logic? It's updated since version 0.2. It might cause strange problem like this issue.

FateScript avatar May 11 '22 09:05 FateScript

Did you check the pre-process logic? It's updated since version 0.2. It might cause strange problem like this issue.

你说的是tensorrt/yolox.cpp的前处理吗?我怀疑是yolox_tiny.pth的问题,因为我用yolox_s.pth转.engine识别效果就不错。

EkilGitHub avatar May 12 '22 00:05 EkilGitHub

注意,tiny的输入是416乘416,要在yolox.cpp里面修改

wangxiaolin-cumt avatar Jun 16 '22 02:06 wangxiaolin-cumt

注意,tiny的输入是416乘416,要在yolox.cpp里面修改

好的,谢谢回复。因为我这边训练时候资源有限,识别的时候选用的是832 x 832的尺寸进行的推理。 是否用TensorRT/C++推理的时候,识别与训练尺寸必须保证严格一致?

EkilGitHub avatar Jun 23 '22 09:06 EkilGitHub