search

fangpin / siamese-pytorch

Implementation of Siamese Networks for image one-shot learning by PyTorch, train and test model on dataset Omniglot
279 stars 59 forks source link

Is there any inference only script? #9

Closed bemoregt closed 4 years ago

bemoregt commented 4 years ago

Hi, @urbanophile @fangpin

Is there any inference only script?

I want to enter just two image for to get similarity result.

Thanks. Best, @bemoregt.

urbanophile commented 4 years ago

Not in this repo, sorry! In any event, there is no pretrained weights either.

fangpin commented 4 years ago

Sorry, I can not access the server to retrieve the model weights, since I just graduated recently. As for the inference only script, I think it's easy for you to follow the Pythorch official tutorial to train the model, save the weights, and load the weights to infer.