LLaVA vs. CoDet

pip install autodistill autodistill-llava autodistill-codet

from autodistill_llava import LLaVA
from autodistill_codet import CoDet

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

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

models = [
    LLaVA(ontology=ontology),
    CoDet(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"
)