search

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

Implementation of Siamese Neural Networks for One-shot Image Recognition
601 stars 180 forks source link

prediction code #7

Closed MansourTrabelsi closed 5 years ago

MansourTrabelsi commented 5 years ago

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

tensorfreitas commented 5 years ago

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.