Using Autodistill, you can compare Grounded SAM and SAM-CLIP on your own images in a few lines of code.
Here is an example comparison:
To start a comparison, first install the required dependencies:
pip install autodistill autodistill-grounded-sam autodistill-sam-clip
Next, create a new Python file and add the following code:
from autodistill_grounded_sam import GroundedSAM
from autodistill_sam_clip import SAMCLIP
from autodistill.detection import CaptionOntology
from autodistill.utils import compare
ontology = CaptionOntology(
{
"solar panel": "solar panel",
}
)
models = [
GroundedSAM(ontology=ontology),
SAMCLIP(ontology=ontology)
]
images = [
"/home/user/autodistill/solarpanel1.jpg",
"/home/user/autodistill/solarpanel2.jpg"
]
compare(
models=models,
images=images
)
Above, replace the images in the `images` directory with the images you want to use.
The images must be absolute paths.
Then, run the script.
You should see a model comparison like this:
When you have chosen a model that works best for your use case, you can auto label a folder of images using the following code:
base_model.label(
input_folder="./images",
output_folder="./dataset",
extension=".jpg"
)
