From ab3f4df93581a2f05c2447e9f17cc2aa0422c6b7 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Mon, 10 Jun 2024 23:46:50 +0200 Subject: [PATCH] feat: use vit_b as default for experiment --- experiments_mask_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiments_mask_generation.py b/experiments_mask_generation.py index 8e32c29b..272adfc5 100644 --- a/experiments_mask_generation.py +++ b/experiments_mask_generation.py @@ -15,7 +15,7 @@ sam_options = SAMOptions( dino_erode_or_dilate=0, dino_debug=False, max_num_boxes=2, - model_type="vit_l" + model_type='vit_b' ) mask_image = generate_mask_from_image(image, sam_options=sam_options)