TensorRT-CenterNet icon indicating copy to clipboard operation
TensorRT-CenterNet copied to clipboard

Published 20 hours ago • verified publisher icon CaoWGG

Different result form pytorch and tensorrt

Open ElonKou opened this issue 4 years ago • 2 comments

Thanks for your great work. I have test the model for pytorch and tensorrt which trained on my dataset. The dataset only have 7 classes. This is result form pytorch model. ok and this the tesorrt model conterted from onnx. error It seems that tensorrt mode has more small object on the top of image, and also have some overlap bbox. Is there any way to solve this problem? thanks!

ElonKou avatar Jun 04 '20 04:06 ElonKou

The small objects disappeared when I changed the classNum and className. But the result still contains overlap. result_screenshot_05 06 2020

ElonKou avatar Jun 05 '20 07:06 ElonKou

I think this is due to no nms implementation as specified in Readme file

janismdhanbad avatar Aug 28 '20 13:08 janismdhanbad