search

hustvl / ViTMatte

[Information Fusion (Vol.103, Mar. '24)] Boosting Image Matting with Pretrained Plain Vision Transformers
MIT License
363 stars 37 forks source link

inference in c++? #18

Open antithing opened 1 year ago

antithing commented 1 year ago

Hi! Thank you for this work. Is there any plan to release c++ inference code, using libtorch / tensorRT / onnx?

Thanks

JingfengYao commented 11 months ago

Sorry, it's not on our plan. Thanks for your attention.

rafaelperez commented 2 months ago

Maybe too late, but I ported VITMatte to compile to Torchscript, which can be easily converted to ONNX for example. https://github.com/rafaelperez/ViTMatte-for-Nuke