TensorRT/C++ 进行推理的时候,出来的结果不对
运用trt.py将yolox_tiny.pth转换成model_trt.pth, model_trt.engine
其中model_trt.pth 预测的结果和yolox_tiny.pth一样:
但是model_trt.engine预测的结果狠奇怪:

请问是啥原因导致的?
Did you check the pre-process logic? It's updated since version 0.2. It might cause strange problem like this issue.
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识别效果就不错。
注意,tiny的输入是416乘416,要在yolox.cpp里面修改
注意,tiny的输入是416乘416,要在yolox.cpp里面修改
好的,谢谢回复。因为我这边训练时候资源有限,识别的时候选用的是832 x 832的尺寸进行的推理。 是否用TensorRT/C++推理的时候,识别与训练尺寸必须保证严格一致?