siriusdemon / Build-Your-Own-Face-Model

Build your own ArcFace, CenterFace/Centernet
334 stars 69 forks source link

Pre-trained model #1

Closed railton closed 4 years ago

railton commented 4 years ago

Could you make the pre-trained model available?

Thanks for the tutorial

siriusdemon commented 4 years ago

Yes, I will upload a facemobilenet pre-trained model.

jyuan1118 commented 4 years ago

Could you make the pre-trained model available?

Thanks for the tutorial

I have uploaded two pretrained models.

face mobilenet accuracy 96.8% https://drive.google.com/drive/folders/1WbIUNF2EAOZNq9eWCTiD_5PtU2_hyh70?usp=sharing

Resnet accuracy: 96% https://drive.google.com/drive/folders/1SyzQ9bPfUXu5EXUE7dlZmsD2pI0M9Os5?usp=sharing

siriusdemon commented 4 years ago

I have uploaded a face-mobile-net model in directory checkpoints

Comedian1926 commented 4 years ago

Thank you for your work, can you tell me the hyperparameters you set, I am on the aligned data set (112, 96), after 20 epochs, the loss has not dropped around 15

siriusdemon commented 4 years ago

All in config.py

Comedian1926 commented 4 years ago

All in config.py

thk