search

KwaiVGI / LivePortrait

Bring portraits to life!
https://liveportrait.github.io
Other
12.8k stars 1.36k forks source link

Using the torchgeometry library and CUDA to replace the CPU version of pasteback function significantly improved the speed #313

Open warmshao opened 2 months ago

warmshao commented 2 months ago

我在FasterLivePortrait 有验证过，速度快不少，可以试一下

https://github.com/warmshao/FasterLivePortrait/blob/981057e7d75aba64e69c070a10391690143ad31b/src/utils/crop.py#L41

cleardusk commented 2 months ago

@warmshao 感谢验证。会比 kornia 更快吗？

warmshao commented 2 months ago

@warmshao 感谢验证。会比 kornia 更快吗？

还没有试过kornia，感觉速度差不多，因为都用上了cuda。但肯定比opencv快😄😄

cleardusk commented 2 months ago

@warmshao https://pypi.org/project/torchgeometry/ 该库似乎已经很早停止维护了，且为 kornia 的 fork 版本

warmshao commented 2 months ago

@warmshao https://pypi.org/project/torchgeometry/ 该库似乎已经很早停止维护了，且为 kornia 的 fork 版本

哦，那这个函数跟kornia应该是等价的，没注意到停止维护了