Skip to content

autodistill/autodistill-mobileclip

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Autodistill MobileCLIP Module

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

MobileCLIP, developed by OpenAI, is a computer vision model trained using pairs of images and text. You can use MobileCLIP with autodistill for image classification.

Read the full Autodistill documentation.

Read the MobileCLIP Autodistill documentation.

Installation

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

pip3 install autodistill-mobileclip

Quickstart

from autodistill_mobileclip import MobileCLIP
from autodistill.detection import CaptionOntology

# define an ontology to map class names to our MobileCLIP 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 = MobileCLIP(
    ontology=CaptionOntology(
        {
            "person": "person",
            "a forklift": "forklift"
        }
    )
)
base_model.label("./context_images", extension=".jpeg")

License

[add license information here]

🏆 Contributing

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

About

MobileCLIP base model for use with Autodistill.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published