Update rynnec/__init__.py

rynnec/__init__.py

@@ -145,7 +145,7 @@ def mm_infer(images_or_videos, vlprocessor, instruct, model, tokenizer, modal='v
 
     return outputs
 
-def mm_infer_segmentation(images_or_videos, vlprocessor, instruct, model, tokenizer, modal='video', seg_start_idx=0, **kwargs):
+def mm_infer_segmentation(images_or_videos, vlprocessor, instruct, model, tokenizer, modal='video', seg_start_idx=0, mask_threshold=0.5, **kwargs):
 
     image2maskids = kwargs.get('image2maskids', [])
     img_size=1024
@@ -264,6 +264,6 @@ def mm_infer_segmentation(images_or_videos, vlprocessor, instruct, model, tokeni
         )
 
     outputs = tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0].strip()
-    pred_masks_sigmoid = pred_masks.sigmoid()>0.5
+    pred_masks_sigmoid = pred_masks.sigmoid() > mask_threshold
 
     return outputs, pred_masks_sigmoid