predict.py

from PIL import Image

from facenet import Facenet
#
# if __name__ == "__main__":
#     model = Facenet()
#     while True:
#         image_1 = input('Input image_1 filename:')
#         try:
#             image_1 = Image.open(image_1)
#         except:
#             print('Image_1 Open Error! Try again!')
#             continue
#
#         image_2 = input('Input image_2 filename:')
#         try:
#             image_2 = Image.open(image_2)
#         except:
#             print('Image_2 Open Error! Try again!')
#             continue
#
#         probability = model.detect_image(image_1,image_2)
#         print(probability)


if __name__ == "__main__":
    model = Facenet()
    image_1 = 'img/1_001.jpg'
    image_1 = Image.open(image_1)
    probability = model.get_embedding(image_1)
    print(probability)