search

tobibaum / metrabs_trt

Running 3D HPE at 30fps
12 stars 3 forks source link

Test-Time Augmentation and Implausible Pose Suppression #2

Closed steb6 closed 2 years ago

steb6 commented 2 years ago

Hi, I am using MetrABS for a research project and I find amazing that someone else was trying to speed it up with TensorRT! Thanks for making this repo which allowed me to add the signatures and export the models as Onnx. I am trying to do the same thing that you did, but in Python. I managed to do everything following your instructions (Yolov4 and BackBone with TensorRT, the MetrabsHeads with TensorFlow), but the results that I am achieving are much worse than the one that comes out from model.detect_poses. I suppose that this happens because we are not doing test-time augmentation and implausible pose suppression. The result that you are obtaining are in line with the one of model.detect_poses?

mrpositron commented 2 years ago

Hello @StefanoBerti!

Have you solved this problem?

steb6 commented 2 years ago

Hi @mrpositron, the problem here was that he was not using homography as a preprocessing phase for the backbone. Take a look to my repository where I managed to accelerate the best model of MetrABS at 50 FPS

p.s. the README is not updated, open an issue if you need anything