Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

希望提供训练hivision_modnet的代码 #190

Open
gnodgnodtonmi opened this issue Oct 12, 2024 · 1 comment
Open

希望提供训练hivision_modnet的代码 #190

gnodgnodtonmi opened this issue Oct 12, 2024 · 1 comment

Comments

@gnodgnodtonmi
Copy link

感谢作者们的出色工作!

在可选的 matting 模型中,BiRefNet 效果最好,但是其模型参数量和结构使得它在性能较差的机器上运行起来非常慢(即使训练一个参数量更小的模型并使用一些其他模型加速方法)。

而在其余模型里,hivision_modnet 是性能效果和速度最均衡的,模型结构比起 BiRefNet 也更容易做量化等处理。
但可能因为模型参数量较小或训练不充分,抠图效果仍然不太能令人满意。

我想训练一个更大模型参数量/效果更好的 hivision_modnet
请问作者是否考虑有可能提供训练 hivision_modnet 的代码(包括数据集)呢?

@Levin-l1ang
Copy link

附议,briefnet远要比cpu的几个网络慢得多

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants
@gnodgnodtonmi @Levin-l1ang and others