CaoWGG / TensorRT-CenterNet

tensorrt5 , centernet , centerface, deform conv, int8
767 stars 158 forks source link

Quite different bbox size between onnx and tensorrt #60

Open Harryqu123 opened 4 years ago

Harryqu123 commented 4 years ago

Hi, I run the centerface onnx model and find although faces are detected the bbox size are quite different between onnx model and tensorrt model. out result You can see that the first image is the one I run use the given onnx model and the second image is the result I run with the converted tensorrt engine. Anyone find similar issue and know how to solve this?

SthPhoenix commented 4 years ago

A bit late, but if it might help, I had same bug with TensorRT 7.0 (on nvidia TensorRT 20.03 and Triton Inference Server 20.03 images). Updating to latest TensorRT (on 20.09 image) fixed the problem