SigLIP base model for use with Autodistill
Project description
Autodistill SigLIP Module
This repository contains the code supporting the SigLIP base model for use with Autodistill.
CLIP, developed by OpenAI, is a computer vision model trained using pairs of images and text. You can use CLIP with autodistill for image classification.
Read the full Autodistill documentation.
Read the SigLIP Autodistill documentation.
Installation
To use SigLIP with autodistill, you need to install the following dependency:
pip3 install autodistill-clip
Quickstart
from autodistill_siglip import SigLIP
from autodistill.detection import CaptionOntology
# define an ontology to map class names to our SigLIP 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
labels = ["person", "a forklift"]
base_model = SigLIP(
ontology=CaptionOntology({item: item for item in labels})
)
results = base_model.predict("image.jpeg", confidence=0.1)
top_1 = results.get_top_k(1)
# show top label
print(labels[top_1[0][0]])
# label folder of images
base_model.label("./context_images", extension=".jpeg")
License
The SigLIP model is licensed under an Apache 2.0 license.
🏆 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
Close
Hashes for autodistill_siglip-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2815075ea1fb39b21c0a18246285efebcc260e7b5695f5f5b7853cdca9d77c95 |
|
MD5 | 092d5c91f01143ddebfbd242e6f6d687 |
|
BLAKE2b-256 | 1a7b016a0add64cb005eab853d3370f68e0ffc2a4d0b200f925e83cff5f8f961 |