Skip to content

fsbatista/autodistill-grounding-dino

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Autodistill Grounding DINO Module

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

Grounding DINO is a zero-shot object detection model developed by IDEA Research. You can distill knowledge from Grounding DINO into a smaller model using Autodistill.

Read the Grounding DINO Autodistill documentation.

Installation

To use the Grounding DINO base model, you will need to install the following dependency:

pip3 install autodistill-grounding-dino

Quickstart

from autodistill_grounding_dino import GroundingDINO
from autodistill_yolov8 import YOLOv8


# define an ontology to map class names to our GroundingDINO 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 = GroundingDINO(ontology=CaptionOntology({"shipping container": "container"}))

# label all images in a folder called `context_images`
base_model.label("./context_images", extension=".jpeg")

License

The code in this repository 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!

About

Grounding DINO module for use with Autodistill.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 88.5%
  • Makefile 11.5%