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.

Sign up for GitHub

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

有pytorch版本的预训练权重吗？ #6

Open

TongMeng-AI opened this issue May 21, 2024 · 1 comment

TongMeng-AI commented May 21, 2024

No description provided.

Owner

SunnyHaze commented May 21, 2024

你好，感谢你的关注。
原论文使用了私有的巨大篡改数据集训练，如果需要相应的权重，可以参考这个仓库的复现，他将原论文的参数迁移到了pytroch中，可以直接使用。https://github.com/RonyAbecidan/ManTraNet-pytorchhttps://github.com/RonyAbecidan/ManTraNet-pytorch

本仓库没有对齐原仓库的实现，无法严格迁移，目前我也只在CASIAv2和NIST上预训练过本模型，仅供科研参考。

如果还有其他问题，欢迎讨论！

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment