search

ZHKKKe / MODNet

A Trimap-Free Portrait Matting Solution in Real Time [AAAI 2022]
Apache License 2.0
3.84k stars 636 forks source link

Sharing a TensorRT implementation of MODNet #107

Closed jkjung-avt closed 3 years ago

jkjung-avt commented 3 years ago

@ZHKKKe Thanks for sharing your inference code and the pre-trained model.

I took your pre-trained PyTorch model and was able to build a TensorRT engine from it. I shared my TensorRT implementation at jkjung-avt/tensorrt_demos, or more specifically, Demo #7: MODNet. Feel free to give it a try.

ZHKKKe commented 3 years ago

@jkjung-avt It is wonderful! Thanks! I hope to integrate it into this Github project after I check it. As I am busy these two weeks, I will add it later. See what you think.

jkjung-avt commented 3 years ago

I hope to integrate it into this GitHub project after I check it.

I put my code under MIT License. You are welcome to integrate the code into this repo by following the license.

Thank you again for sharing the model.

ZHKKKe commented 3 years ago

Hey, I referenced your project in our README. Thanks again.