tensorrt

[chh7411898]

Did you try to convert this into tensorrt？

[stefanopini]

No, I didn't, but any contributions are welcome! Please consider sharing the results here if you manage to convert it in tensorrt.

[gpastal24]

#14 I have added TRT support for HigherHRNet. I have used the yolov5 conversion function since torch2trt was throwing memory error. It works only on the w32_512 weights

[stefanopini]

Amazing, thanks a lot for the contribution! I will review the PR in the next weeks and consider merging it into the master branch

[gpastal24]

OK , thank you I have made some improvements to the code in order to be better understood.

[stefanopini]

Hi @gpastal24 , unfortunately I didn't have the time to look into this the last two week. I will try to review it by the end of this week, apologies for the waiting

[gpastal24]

Hi @stefanopini, don't worry about it, I am not in a hurry. Thanks for letting me know :)

[stefanopini]

Hi @gpastal24 ! As mentioned in #14 I merged your PR that adds the support to TensorRT using yolov5 to a new branch tensorrt. Thanks a lot for the help with the code!

I'm now working on the PR #15 to clean up the code a bit, update the documentation (and hopefully add a link to a working example on colab) and then merge it into the master branch. I'll let you know when everything is ready, happy to have the pr reviewed by you if you're available.

@chh7411898 the documentation is not there yet, but you can use the tensorrt convertor already from the branch tensorrt! Just follow the instruction posted here

[stefanopini]

TensorRT support has been merged into master! Closing