search

YuzhouPeng / unet-tensorrt

this is a tensorrt version unet, inspired by tensorrtx
MIT License
36 stars 11 forks source link

Can use python? #7

Open missbook520 opened 3 years ago

missbook520 commented 3 years ago

Thank you very much for your work, can I use python to call it?

YuzhouPeng commented 3 years ago

Yes, you need to compile code to a library (.so file) and use pybind11 to call the function. Maybe I will add plugin in the future if I have time :)

missbook520 commented 3 years ago

Thank you very much for your reply. I followed your steps when compiling this project on jeston nano, but the following error was reported when using unet -s. What is the problem? My CUDA version is 10.2, tensorrt version is 7.1.3-1, thank you very much 38b6ea7cd23228a846a9a65a60f089b

YuzhouPeng commented 3 years ago

你有使用链接里的pytorch unet训练权重吗？我看了下好像weights翻倍了

YuzhouPeng commented 3 years ago

如果训练按照步骤来的话，可能是factor的问题，我抽时间fix下把

YuzhouPeng commented 4 months ago

https://github.com/wang-xinyu/tensorrtx/tree/master/unet please use newest repo for testing, this old repo is no longer maintained