facenet_pytorch icon indicating copy to clipboard operation
facenet_pytorch copied to clipboard

Pre-Trained Models

Open ahkarami opened this issue 7 years ago • 7 comments

Dear @liorshk, Thank you for your code. Do you have any pre-trained models? Would you please, release the weights of your pre-trained models?

ahkarami avatar Feb 02 '18 11:02 ahkarami

Hi, have you found the pre-trained model or a good way to convert from the tensorflow model to pytorch?

xiaoiker avatar Jul 24 '18 13:07 xiaoiker

@xiaoiker You can use MMdnn or SphereFace.

ahkarami avatar Jul 24 '18 15:07 ahkarami

@ahkarami @xiaoiker I download the pretrain model from https://github.com/davidsandberg/facenet (20180402-114759) Then I install mmdnn Then I run this code mmconvert -sf tensorflow -in model-20180402-114759.meta -df pytorch -om 20180402-114759.pth Error: image

Am I writing it wrong?

@xiaoiker You can use MMdnn or SphereFace.

chungyau97 avatar Apr 03 '19 06:04 chungyau97

Dear @chungyau97, Thank you for your effort. Really, I didn't do that (i.e., converting the model). You can send an Issue about it in the MMdnn GitHub repository.

ahkarami avatar Apr 03 '19 07:04 ahkarami

@ahkarami I'm curious about the SphereFace. Why do you recommend this?

chungyau97 avatar Apr 03 '19 10:04 chungyau97

It is just a suggestion. You can refer to its paper for more information.

ahkarami avatar Apr 03 '19 10:04 ahkarami

@xiaoiker @chungyau97 there is a pytorch conversion of the two pretrained inception resnet v1 models (vggface and casia-web face) from davidsandberg/facenet here:

https://github.com/timesler/facenet-pytorch

Disclaimer: it is my repo.

timesler avatar Jun 03 '19 02:06 timesler