OWL ViT vs. DETIC

pip install autodistill autodistill-owl-vit autodistill-detic

from autodistill_owl_vit import OWLViT
from autodistill_detic import DETIC

from autodistill.detection import CaptionOntology
from autodistill.utils import compare

ontology = CaptionOntology(
    {
        "solar panel": "solar panel",
    }
)

models = [
    OWLViT(ontology=ontology),
    DETIC(ontology=ontology)
]

images = [
    "/home/user/autodistill/solarpanel1.jpg",
    "/home/user/autodistill/solarpanel2.jpg"
]

compare(
    models=models,
    images=images
)

base_model.label(
  input_folder="./images",
  output_folder="./dataset",
  extension=".jpg"
)