Bigger number of slots

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):