def compute_gallery_embeddings(self):
''' Computes embedding vectors for the gallery. '''
images = []
for full_path in tqdm(self.impaths, desc='Reading gallery images.'):
image = cv2.imread(full_path)
if image is None:
log.error("Cannot find image, full_path =", full_path)
image = crop_resize(image, self.input_size)
images.append(image)
embeddings = [None for _ in self.impaths]
index = 0
for image in tqdm(images, desc='Computing embeddings of gallery images.'):
embeddings[index] = self.model.predict(image).reshape([-1])
index += 1
return embeddings
def compute_embedding(self, image):
''' Takes input image and computes embedding vector. '''
image = crop_resize(image, self.input_size)
embedding = self.model.predict(image)
return embedding