Zero Shot Object Detection Example

This folder contains an example script which demonstrates the usage of OWL-ViT to run zero shot object detection task on Gaudi platform.

Single-HPU inference

python3 run_example.py \
    --model_name_or_path google/owlvit-base-patch32 \
    --image_path "http://images.cocodataset.org/val2017/000000039769.jpg" \
    --prompt "a photo of a cat, a photo of a dog" \
    --use_hpu_graphs \
    --bf16

Model that have been validated:

• google/owlvit-base-patch32