| import tensorflow as tf | |
| def load_image(image_path): | |
| img = tf.io.read_file(image_path) | |
| img = tf.image.decode_jpeg(img, channels=3) | |
| img = tf.image.resize(img, (299, 299)) | |
| img = tf.keras.applications.inception_v3.preprocess_input(img) | |
| return img, image_path | |
| # Feature extraction | |
| image_model = tf.keras.applications.InceptionV3(include_top=False, | |
| weights='imagenet') | |
| new_input = image_model.input | |
| hidden_layer = image_model.layers[-1].output | |
| image_features_extract_model = tf.keras.Model(new_input, hidden_layer) | |
| # Tokenizer | |