feat: use vit_b as default for experiment

2024-06-10 23:46:50 +02:00 · 2024-06-10 23:46:50 +02:00 · ab3f4df935
parent ead1ed617a
commit ab3f4df935
1 changed files with 1 additions and 1 deletions
--- 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)