yobibyte / yobiface

Usage of https://github.com/davidsandberg/facenet/ model for people identification
MIT License
65 stars 41 forks source link

Train on my own dataset #2

Open Zumbalamambo opened 7 years ago

Zumbalamambo commented 7 years ago

How do I train on my own dataset?

yobibyte commented 7 years ago

Go here for training. I used the pretrained model from the same repo.

yeluohuazhifou commented 6 years ago

when using your code,reported this error.

NotFoundError: Tensor name "incept3b/in2_conv3x3_12/batch_norm/cond/incept3b/in2_conv3x3_12/batch_norm/moments/moments_1/variance/ExponentialMovingAverage/local_step" not found in checkpoint files ./model_check_point/model-20160506.ckpt-500000

Have you encountered this problem?

yobibyte commented 6 years ago

Sorry for late reply, I fixed the issue, you need to download the new model though.

yeluohuazhifou commented 6 years ago

@yobibyte thank you,i will try it later.But i read the file you sent , there is a .pb file i did not see it when i was training the model.

yobibyte commented 6 years ago

I think, you do not need .pdb file. You just load a meta graph from .meta file and then restore the model from a checkpoint.

lidongxing commented 6 years ago

NotFoundError (see above for traceback): Key affine1/biases not found in checkpoint