bubbliiiing / arcface-pytorch

这是一个arcface-pytorch的源码,可以用于训练自己的模型。
MIT License
124 stars 21 forks source link

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

Open zhusai2401 opened 1 year ago

zhusai2401 commented 1 year ago

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

bubbliiiing commented 1 year ago

貌似annotation_path 对了就可以把

zhusai2401 commented 1 year ago

貌似annotation_path 对了就可以把

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

zhusai2401 commented 1 year ago

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

zhusai2401 commented 1 year ago

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

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

bubbliiiing commented 1 year ago

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