Can use python?

[missbook520]

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

[YuzhouPeng]

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]

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

[YuzhouPeng]

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

[YuzhouPeng]

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

[YuzhouPeng]

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