INNER CODE UNIT · Python

transform_func

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

    transform_func = create_transform(**config)
    dataset = LogoDataset(
        root_dir, transform_func, split_set
    )  # our dataset is built on the data written in val.txt
    data_loader = DataLoader(dataset, batch_size, num_workers=2)
    embeddings, labels, elapsed = generate_embeddings(model, data_loader, device)
    return embeddings, labels, dataset, elapsed


def run_clip_model(
    root_dir: Path,
    split_set: Optional[Set[str]],
    model_name: str,
    batch_size: int,
    device: torch.device,
):
    model = CLIPModel.from_pretrained(f"openai/{model_name}").vision_model
    model.to(device)

View source record →