Skip to content

SuperPoint TensorRT implementation#1034

Merged
wang-xinyu merged 2 commits intowang-xinyu:masterfrom
xiang-wuu:tensorrtx
Jul 6, 2022
Merged

SuperPoint TensorRT implementation#1034
wang-xinyu merged 2 commits intowang-xinyu:masterfrom
xiang-wuu:tensorrtx

Conversation

@xiang-wuu
Copy link
Copy Markdown
Contributor

Code in this PR implements baseline PyTorch NN using TensorRT Network API, also provide instructions to run demo script by inferring TensorRT generated engine file along with pre-trained PyTorch weight model, the updated code base of original magicleap/SuperPointPretrainedNetwork is maintained in xiang-wuu/SuperPointPretrainedNetwork

@wang-xinyu
Copy link
Copy Markdown
Owner

wang-xinyu commented Jul 5, 2022

@xiang-wuu Thanks for the PR!

Can you please share one output image generated from your code?

And does this support tensorrt8.x or 7.x?

Do you have the inference time comparison between pytorch-gpu and your tensorrt implementation?

@xiang-wuu
Copy link
Copy Markdown
Contributor Author

@wang-xinyu

Updated the README for sample output result, it's almost identical for PyTorch and TensorRT implementation.

I have tested it only with tensorrt7.x but it should also work with tensorrt8.x as well!

I haven't done the inference time comparison but i put it as separate task under speed/accuracy validation as future work.

@wang-xinyu
Copy link
Copy Markdown
Owner

@xiang-wuu Thanks a lot! merging it.

BTW, the 8.x has slightly different APIs from 7.x, please do consider to test with 8.x in the future work if you got any chance.

@wang-xinyu wang-xinyu merged commit fdd1364 into wang-xinyu:master Jul 6, 2022
@xiang-wuu
Copy link
Copy Markdown
Contributor Author

@xiang-wuu Thanks a lot! merging it.

BTW, the 8.x has slightly different APIs from 7.x, please do consider to test with 8.x in the future work if you got any chance.

Sure!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants