yinguobing/cnn-facial-landmark

Using the pre trained model to train my own dataset ( transfer learning )

Opened this issue · 5 comments

I have trained my own dataset and I have created my own model from scratch but it doesn't give good results.
The problem here that eye region landmarks = 40 points and for pre-trained model 68 points ( facial region ).
I think that is possible to train my own dataset from the pre-trained model that you give it using "Transfer learning", but I don't have any idea how to use this technique on the pre-trained model that you give.
Do you have any idea how can I use that?

TensorFlow's official model shows how to do transfer learning here: https://github.com/tensorflow/models/blob/master/official/resnet/resnet_run_loop.py

Pay attention to these keywords: pretrained_model_checkpoint_path and fine_tune.

Thanks.
This is really so complicated, but I will try to use this script.
Can you tell me what function should I use?

What do you think in my case for fine_tune?
Should I make false or change it to true?

I think it should be set to True but you need to figure this out through reading the code. It's been a while since the last time I read it.

Thanks.
I will see it.