search

peteryuX / arcface-tf2

ArcFace unofficial Implemented in Tensorflow 2.0+ (ResNet50, MobileNetV2). "ArcFace: Additive Angular Margin Loss for Deep Face Recognition" Published in CVPR 2019. With Colab.
MIT License
260 stars 60 forks source link

SIze of model #13

Closed seunboy1 closed 4 years ago

seunboy1 commented 4 years ago

Thank you for this project. I would like to know if there is a way to reduce the size of the model using MobileNetV2 as backbone architecture.

peteryuX commented 4 years ago

Hi @seunboy1 ~ The input argment alpha in the tf.keras.applications.MobileNetV2() is the thing you want~ (detail in here).

seunboy1 commented 4 years ago

Ok, thank you

seunboy1 commented 4 years ago

Hi how do i create a pb file from the checkpoint

peteryuX commented 4 years ago

Call tf.keras.models.save_model(model, "filepath_you_want") in test.py.