Fine-tuning and retraining of the model

[andergalvaan]

Hi,

Does anyone know of a method to retrain the model with some images and then test it with images of the same and different people? I mean, it is possible to keep the feature extraction that the model implements, but retraining the model to better detect some people? I have retrained the model with images of some people and it detects them more reliably through cosine similarity, but with images of people who do not participate in the retraining more false positives appear.

Thank you in advanced,
Ander

[Moizcode]

how did you fine-tune, have you used --pretrained_model flag, and if yes what path you give?

[Shun7200101]

@andergalvaan
你可以参考 train_softmax.py 中的迁移学习示例：

python src/train_softmax.py --pretrained_model models/pretrained/20180402-114759 --data_dir your_data