INNER CODE UNIT · Python

__getitem__

openfoodfacts/openfoodfacts-ai · logo-ann/benchmarks/embedding_models_benchmark/main.py:67

    def __getitem__(self, idx: int):
        image = Image.open(self.image_paths[idx])
        return (self.transform_func(image), self.labels[idx])

    def __len__(self):
        return len(self.image_paths)


def generate_embeddings(
    model: torch.nn.Module, data_loader: DataLoader, device: torch.device
) -> Tuple[torch.Tensor, torch.Tensor, float]:
    model.eval()
    embedding_all = []
    labels_all = []
    elapsed = 0.0

    with torch.inference_mode():
        for inputs, labels in tqdm.tqdm(data_loader):

View source record →