arcface-pytorch icon indicating copy to clipboard operation
arcface-pytorch copied to clipboard

请问,如果我想使用MS_Celeb_1M数据集训练,需要修改哪些内容?

Open zhusai2401 opened this issue 3 years ago • 5 comments

请问,如果我想使用MS_Celeb_1M数据集训练,是否只要把net.arcface.py里的embedding_size和 num_classes更改就可以了?还有你的arcface_training.py的作用是什么呢~~

zhusai2401 avatar Nov 09 '22 05:11 zhusai2401

貌似annotation_path 对了就可以把

bubbliiiing avatar Nov 09 '22 15:11 bubbliiiing

貌似annotation_path 对了就可以把

num_classes不是图片数量吗?我看你写的num_classes是10575,是webface的图片数量,换别的数据集,图片的ID数量会变动,这时,num_classes不需要改成自己数据集的数量吗?

zhusai2401 avatar Nov 10 '22 01:11 zhusai2401

貌似annotation_path 对了就可以把 还有,大佬,我用你代码,使用arcface官网提供的webface进行训练,backbone选择的是mobilefacenet,但是iters的速度是1-2it/s,我是在ubuntu系统,用的GPU是GP100,请问是什么原因呢?GPU利用率一直从0开始波动,没有稳定状态。是我电脑有问题吗?

zhusai2401 avatar Nov 10 '22 07:11 zhusai2401

貌似annotation_path 对了就可以把还有了 ,各位,我用你代号,使用arcface官网提供的webface进行训练,骨干选择的是mobilefacenet,但是iters的速度是1-2it/s,我在ubuntu系统,使用的GPU是GP100,请问是什么原因呢?GPU利使用率一直从0开始波动,没有稳定状态。是我电脑有问题吗?

已经解决,是因为我是在机械硬盘上运行,CPU不好导致的,把数据集移到固态上,速度快了起来。

zhusai2401 avatar Nov 15 '22 03:11 zhusai2401

只要txt有了,会自适应算合适的num classes

bubbliiiing avatar Nov 15 '22 15:11 bubbliiiing