BabitMF / bmf

Cross-platform, customizable multimedia/video processing framework. With strong GPU acceleration, heterogeneous design, multi-language support, easy to use, multi-framework compatible and high performance, the framework is ideal for transcoding, AI inference, algorithm integration, live video streaming, and more.
https://babitmf.github.io/
Apache License 2.0
733 stars 60 forks source link

Inpaint module #59

Closed xiaoweiw-nv closed 9 months ago

xiaoweiw-nv commented 10 months ago

Add image inpaint module using the MAT model.

The inference is run by PyTorch, I tried to export the model to onnx but the implementation contains several plugins so onnx does not work. I literally wrote the module in an afternoon : ) This module should be a good demo to show how fast the developer can integrate a PyTorch model into BMF.