search

advimman / lama

🦙 LaMa Image Inpainting, Resolution-robust Large Mask Inpainting with Fourier Convolutions, WACV 2022
https://advimman.github.io/lama-project/
Apache License 2.0
7.62k stars 821 forks source link

ONNX Model done #315

Open OPHoperHPO opened 2 months ago

OPHoperHPO commented 2 months ago

Hello everyone,

We at @Carve-Photos have successfully ported LaMa (big-lama) to ONNX with results closely resembling the original. Check it out here: https://huggingface.co/Carve/LaMa-ONNX

Upd. Here is HG Space that uses our model https://huggingface.co/spaces/Carve/LaMa-Demo-ONNX

Particle1904 commented 2 months ago

Thank you so much for this! I tried to convert it but failed miserably. Its always a bummer when you find a extremely useful model that can only be used through Python/C/C++... makes it so hard to use it on custom built tools.

senya-ashukha commented 2 months ago

Amazing work! Do you want send a pool request mentioning that in the top of the readme page?

OPHoperHPO commented 2 months ago

Amazing work! Do you want send a pool request mentioning that in the top of the readme page?

We are currently preparing an update to speed up the onnx model. When we release it, we will submit a pull request with guide on how to export the model.

OPHoperHPO commented 2 months ago

@senya-ashukha We have released an updated onnx model and submitted PR #316