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