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

pre-generated file的相关问题 #1

Open
mengsiwei opened this issue Nov 13, 2021 · 5 comments
Open

pre-generated file的相关问题 #1

mengsiwei opened this issue Nov 13, 2021 · 5 comments

Comments

@mengsiwei
Copy link

请问,训练前需要对数据进行处理的方法有哪些呀?可以提供一份你的datasets吗,我的邮箱是mengsiwei2021@email.szu.edu.cn

@PeiwenSun2000
Copy link
Owner

这里暂且只对ferplus数据集进行了复现,ferplus正如论文所说已经进行了一定程度上的对齐,所以可以跳过对齐步骤。

ferplus数据集包括Train, Valid和Test图片以及一些标注文件txt,这里和原作者做法一样使用Train+Valid 进行训练。具体数据集的下载建议去官方网站下载。

用mv2folder.py对下载下来的数据集进行裁减和resize并且保存至对应文件夹,预处理就可以了

@mengsiwei
Copy link
Author

你好,请问训练时候的backbone.pth有提供吗?

@PeiwenSun2000
Copy link
Owner

backbone.pth是iresnet系列的预训练模型可以从arcface训练得来

建议可以先从resnet18预训练模型开始 ijba from Kai Wang,如果有机会可以从MSCeleb进行预训练可以得到更好的效果

@mengsiwei
Copy link
Author

你好,没有用到resnet18模型进行训练,test的时候出现模型不匹配问题,有什么解决方案呢。还想请问是必须用多台GPU训练吗?

@PeiwenSun2000
Copy link
Owner

善用argparse.ArgumentParser传入参数,python不太了解可以多查一查,test也一样。

resnet18 使用2块1080ti的batchsize是64,你可以自己调整,建议batchsize不要低于16.

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

2 participants