Skip to content

Instantly share code, notes, and snippets.

@aanwark
Created August 6, 2019 10:58
Show Gist options
  • Save aanwark/df5fd01700a34d8b6e969414cb76bbea to your computer and use it in GitHub Desktop.
Save aanwark/df5fd01700a34d8b6e969414cb76bbea to your computer and use it in GitHub Desktop.
import cv2
from train import train
from keras.models import load_model
t = train()
img_left = cv2.imread("../data/1.bmp")
img_right = cv2.imread("../test_data/0.jpg")
model = t.get_siamese_model()
model.load_weights("siamese.h5")
prediction = t.predict(model, img_left, img_right)
print("Prediction: ", prediction)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment