ManTraNet-Pytorch icon indicating copy to clipboard operation
ManTraNet-Pytorch copied to clipboard
verified publisher icon SunnyHaze

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

Open TongMeng-AI opened this issue 1 year ago • 1 comments

TongMeng-AI avatar May 21 '24 01:05 TongMeng-AI

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

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

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

SunnyHaze avatar May 21 '24 04:05 SunnyHaze