Skip to main content

OWLv2 base model for use with Autodistill.

Project description

Autodistill OWLv2 Module

This repository contains the code supporting the OWLv2 base model for use with Autodistill.

OWLv2 is a zero-shot object detection model that follows from on the OWL-ViT architecture. OWLv2 has an open vocabulary, which means you can provide arbitrary text prompts for the model. You can use OWLv2 with autodistill for object detection.

Read the full Autodistill documentation.

Read the OWLv2 Autodistill documentation.

Installation

To use OWLv2 with autodistill, you need to install the following dependency:

pip3 install autodistill-owlv2

Quickstart

from autodistill_owlv2 import OWLv2
from autodistill.detection import CaptionOntology

# define an ontology to map class names to our OWLv2 prompt
# the ontology dictionary has the format {caption: class}
# where caption is the prompt sent to the base model, and class is the label that will
# be saved for that caption in the generated annotations
# then, load the model
base_model = OWLv2(
    ontology=CaptionOntology(
        {
            "person": "person",
            "a forklift": "forklift"
        }
    )
)

# run inference on a single image
results = base_model.predict("./context_images/image.png")

base_model.label("./context_images", extension=".jpeg")

License

This model is licensed under an Apache 2.0 (see original model implementation license, and the corresponding HuggingFace Transformers documentation).

🏆 Contributing

We love your input! Please see the core Autodistill contributing guide to get started. Thank you 🙏 to all our contributors!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

autodistill-owlv2-0.1.1.tar.gz (7.5 kB view details)

Uploaded Source

Built Distribution

autodistill_owlv2-0.1.1-py3-none-any.whl (7.8 kB view details)

Uploaded Python 3

File details

Details for the file autodistill-owlv2-0.1.1.tar.gz.

File metadata

  • Download URL: autodistill-owlv2-0.1.1.tar.gz
  • Upload date:
  • Size: 7.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.11.4

File hashes

Hashes for autodistill-owlv2-0.1.1.tar.gz
Algorithm Hash digest
SHA256 ee823e4cc7ed07e74412c73c2a454f1c1f5a61c9651b27aeeb672b172901ebb2
MD5 43c05bf362f3175007803aa2e0e19cc7
BLAKE2b-256 db37fc49346158940fd1281f93e5169896495bf5d593c507c0335edeb5bf509d

See more details on using hashes here.

File details

Details for the file autodistill_owlv2-0.1.1-py3-none-any.whl.

File metadata

File hashes

Hashes for autodistill_owlv2-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 1d2c564849e77bcbfb07b2a6ee29c50f282f7256de6c09662d93bc43eb80054e
MD5 bea61a93f2066a0221163cc307a2e1c9
BLAKE2b-256 5d0e1c7e3a06b7fd78f37d6f8da1c6cb863a4e0c28724e17c6cc2442c68c970c

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page