autodistill / autodistill-detic

DETIC module for use with Autodistill.
https://docs.autodistill.com
MIT License
13 stars 2 forks source link
autodistill computer-vision detic

Autodistill DETIC Module

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

DETIC is a transformer-based object detection and segmentation model developed by Meta Research.

Read the full Autodistill documentation.

Read the DETIC Autodistill documentation.

Installation

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

pip3 install autodistill-detic

Quickstart

from autodistill_detic import DETIC

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

License

The code in this repository is licensed under an MIT license.

See the Meta Research DETIC repository for more information on the DETIC license.

🏆 Contributing

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