TensorRT Edition - Githubissues

lhwcv / mlsd_pytorch

Pytorch implementation of "M-LSD: Towards Light-weight and Real-time Line Segment Detection"

Apache License 2.0

189 stars 37 forks source link

TensorRT Edition #15

Closed rhysdg closed 2 years ago

rhysdg commented 2 years ago

Added TensorRT/ Jetson support with a command line script.
FP16 and int8 calibration support.
Prerequisites, and usage has been addded to the readme.
Tested and benchmarked on both a Xavier NX , and an AGX Xavier (benches have been added to the README).
Added a floor divide update for the latest version of torch.

lhwcv commented 2 years ago

Great work!

rhysdg commented 2 years ago

Cheers! There's a couple more tweaks that need to be added to switch between tiny, and large I've noticed but I'll send the pr your way when I get round to it. Love the work and looking forward to seeing where it goes in the future!