code: def test_mantis_mask_rcnn_predict_from_dl_threshold(
      sample_dataset, pretrained_state_dict
  ):
      model = mask_rcnn.model(num_classes=91)
      model.load_state_dict(pretrained_state_dict)
  
      infer_dl = mask_rcnn.infer_dl(
          dataset=sample_dataset, batch_size=2
      )
      preds = mask_rcnn.predict_from_dl(
          model=model,
          infer_dl=infer_dl,
          show_pbar=False,
          detection_threshold=1.0,
      )
  
      assert len(preds[0].pred.detection.label_ids) == 0
  
file_path: icevision/tests/models/torchvision_models/mask_rcnn/test_predict.py
imports: import pytest
  from icevision.imports import *
  from icevision import *
  from icevision.models.torchvision import mask_rcnn
label_desc: Use image node_id 6 for calling a global function with example usage: test_mantis_mask_rcnn_predict_from_dl_threshold(sample_dataset, pretrained_state_dict) without return types