From 26447806c44c8499cf58f1ce560486b13fea9951 Mon Sep 17 00:00:00 2001 From: alexcbb <alexchapin@hotmail.fr> Date: Thu, 29 Jun 2023 11:42:07 +0200 Subject: [PATCH] Bigger number of slots --- osrt/encoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/osrt/encoder.py b/osrt/encoder.py index 64f61dd..a96650e 100644 --- a/osrt/encoder.py +++ b/osrt/encoder.py @@ -120,9 +120,9 @@ class FeatureMasking(nn.Module): pred_iou_thresh=0.88, points_per_batch=64, min_mask_region_area=0, - num_slots=10, + num_slots=32, slot_dim=1536, - slot_iters=1, + slot_iters=3, sam_model="default", sam_path="sam_vit_h_4b8939.pth", randomize_initial_slots=False): -- GitLab