tensorfreitas / Siamese-Networks-for-One-Shot-Learning

Implementation of Siamese Neural Networks for One-shot Image Recognition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

prediction code

MansourTrabelsi opened this issue · comments

Hi,
I am a beginner in this field and I am really curious about one shot learning because I am facing the lack of data problem. So can you tell me how to use the resulted model to do the prediction.

Thank you

Hi,

To use the model you will just need to frame the prediction as a one-shot task: Assume that you have 5 classes: give the model 5 pairs of images (one to be predicted and the reference class images). The pair with the highest score will be the predicted class.

Hope this is enough to understand the prediction part.

Feel free to reopen if needed.