search

AISangam / Facenet-Real-time-face-recognition-using-deep-learning-Tensorflow

Facenet-Real-time-face-recognition-using-deep-learning-Tensorflow
158 stars 104 forks source link

How to choose a model?(pb file) #2

Closed ytliang97 closed 4 years ago

ytliang97 commented 6 years ago

Hi, I'm trying to use your code. But I stocked at "Training of Model" step. I don't know what model I can choose to put it inside the folder named as “model”. If I want to use other's cnn model for this, would I need to check the input size and bench to fit the pre_img?

peshmerge commented 6 years ago

You can download the mode files from the repo of facenet of David Sandberg https://github.com/davidsandberg/facenet/wiki#pre-trained-models

  1. Choose one model and download it.
  2. Extract the content
  3. Then include the path of the .pb file in your code here https://github.com/AISangam/Facenet-Real-time-face-recognition-using-deep-learning-Tensorflow/blob/d5c2185555f1b107bd908cc9fd12d41b25cd11ae/classifier_train.py#L10
SaddamBInSyed commented 5 years ago

@peshmerge Thanks for the clarification

ytliang97 commented 4 years ago

@peshmerge Thank you!