search

IrvingMeng / MagFace

MagFace: A Universal Representation for Face Recognition and Quality Assessment, CVPR2021, Oral
Apache License 2.0
626 stars 85 forks source link

Train with mobilenet? #19

Closed huytuong010101fpt closed 3 years ago

huytuong010101fpt commented 3 years ago

I see the training code is using ResNetX Can you show me how to train with mobilenet?

IrvingMeng commented 3 years ago

This involves a few steps:

  1. add the script for mobilenet into this foler main/models.
  2. add entrances for mobilenet in load_features: i. https://github.com/IrvingMeng/MagFace/blob/main/models/magface_dist.py#L30-L49 ii. https://github.com/IrvingMeng/MagFace/blob/main/models/magface.py#L22-L41)
  3. change the bash file to use your mobilenet in i. https://github.com/IrvingMeng/MagFace/blob/main/run/run_dist.sh#L12 ii. https://github.com/IrvingMeng/MagFace/blob/main/run/run.sh#L12

Then you are good to go.

huytuong010101fpt commented 3 years ago

This involves a few steps:

  1. add the script for mobilenet into this foler main/models.
  2. add entrances for mobilenet in load_features: i. https://github.com/IrvingMeng/MagFace/blob/main/models/magface_dist.py#L30-L49 ii. https://github.com/IrvingMeng/MagFace/blob/main/models/magface.py#L22-L41)
  3. change the bash file to use your mobilenet in i. https://github.com/IrvingMeng/MagFace/blob/main/run/run_dist.sh#L12 ii. https://github.com/IrvingMeng/MagFace/blob/main/run/run.sh#L12

Then you are good to go.

wow thank u so much, let me try

sfxeazilove commented 10 months ago

Hi, did you get this done and how did you do it?