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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | ee823e4cc7ed07e74412c73c2a454f1c1f5a61c9651b27aeeb672b172901ebb2 |
|
MD5 | 43c05bf362f3175007803aa2e0e19cc7 |
|
BLAKE2b-256 | db37fc49346158940fd1281f93e5169896495bf5d593c507c0335edeb5bf509d |
File details
Details for the file autodistill_owlv2-0.1.1-py3-none-any.whl
.
File metadata
- Download URL: autodistill_owlv2-0.1.1-py3-none-any.whl
- Upload date:
- Size: 7.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.11.4
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1d2c564849e77bcbfb07b2a6ee29c50f282f7256de6c09662d93bc43eb80054e |
|
MD5 | bea61a93f2066a0221163cc307a2e1c9 |
|
BLAKE2b-256 | 5d0e1c7e3a06b7fd78f37d6f8da1c6cb863a4e0c28724e17c6cc2442c68c970c |