search

autodistill / autodistill-grounded-sam

GroundedSAM Base Model plugin for Autodistill
https://docs.autodistill.com
Apache License 2.0
44 stars 9 forks source link

Regarding "how-to-auto-train-yolov8-model-with-autodistill.ipynb" #2

Closed Jordan-Pierce closed 1 year ago

Jordan-Pierce commented 1 year ago

Thanks for this notebook, I apologize if I'm making a ticket in the wrong repo, though they all seem connected. This cell stores the image and annotations for dataset in RAM, and only when it finishes does it output to disk. I'm working with a small video right now so it's not an issue, but have you considered outputting to disk more frequently so RAM doesn't max out? 


from autodistill_grounded_sam import GroundedSAM

base_model = GroundedSAM(ontology=ontology)
dataset = base_model.label(
    input_folder=IMAGE_DIR_PATH, 
    extension=".png", 
    output_folder=DATASET_DIR_PATH)