From 977be10b92511d1356b50baa7cb2deac4d3230ca Mon Sep 17 00:00:00 2001 From: Aroy-Art Date: Mon, 2 Sep 2024 21:34:53 +0200 Subject: [PATCH] Add: finction to handel yes no prompts --- Python/similar-images-classification/main.py | 34 ++++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/Python/similar-images-classification/main.py b/Python/similar-images-classification/main.py index 2cd3eda..efc2dbc 100644 --- a/Python/similar-images-classification/main.py +++ b/Python/similar-images-classification/main.py @@ -6,6 +6,40 @@ from tensorflow.keras.applications.vgg19 import preprocess_input import faiss import cv2 + +def query_yes_no(question, default="yes"): + """Ask a yes/no question via raw_input() and return their answer. + + "question" is a string that is presented to the user. + "default" is the presumed answer if the user just hits . + It must be "yes" (the default), "no" or None (meaning + an answer is required of the user). + + The "answer" return value is True for "yes" or False for "no". + """ + valid = {"yes": True, "y": True, "ye": True, + "no": False, "n": False} + if default is None: + prompt = " [y/n] " + elif default == "yes": + prompt = " [Y/n] " + elif default == "no": + prompt = " [y/N] " + else: + raise ValueError("invalid default answer: '%s'" % default) + + while True: + sys.stdout.write(question + prompt) + choice = input().lower() + if default is not None and choice == '': + return valid[default] + elif choice in valid: + return valid[choice] + else: + sys.stdout.write("Please respond with 'yes' or 'no' " + "(or 'y' or 'n').\n") + + model = vgg19.VGG19(weights="imagenet", include_top=False, pooling="avg")