Pretrained face recognition net in Pytorch

Hi!

So I’m working on a project where my dataset is just people’s images, and I want to finetune it. I need to use a pretrained net which does face recognition (imagenet’s dataset is not similar to my data).

Where can I find a good one? :slight_smile:

Thx in advance!

2 Likes
7 Likes

This project may help you!

6 Likes

I could not find the pytorch weights file can you please share me the pytorch weights model so that i can use it for my project

I’ve ported the popular pretrained tensorflow models from the davidsandberg/facenet Github repo into pretrained pytorch implementations. Pretrained weights are downloaded and loaded into the module on model instantiation, in a manner similar to the torchvision pretrained models. Hope you find it useful!

3 Likes

Really appreciate the efforts given to the project <3