owlvit-base-patch16-torch

Finds any object you name in pictures using 16x16 image patches without needing specific training for those items.

Details

• Model name: owlvit-base-patch16-torch

• Model source: https://huggingface.co/docs/transformers/tasks/zero_shot_object_detection

• Model author: Thomas Wolf, et al.

• Model license: Apache 2.0

• Exposes embeddings? yes

• Tags: detection, logits, embeddings, torch, transformers, zero-shot, official

Requirements

• Packages: torch, torchvision, transformers

• CPU support

  • yes

• GPU support

  • yes

Example usage

import fiftyone as fo
import fiftyone.zoo as foz

dataset = foz.load_zoo_dataset(
    "coco-2017",
    split="validation",
    dataset_name=fo.get_default_dataset_name(),
    max_samples=50,
    shuffle=True,
)


classes = ["person", "dog", "cat", "bird", "car", "tree", "chair"]

model = foz.load_zoo_model(
    "owlvit-base-patch16-torch",
    classes=classes,
)

dataset.apply_model(model, label_field="predictions")

session = fo.launch_app(dataset)