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