code: def select_top_predictions(predictions, confidence_threshold=0.9):
      scores = predictions.get_field("scores").numpy()
      keep = [
          idx
          for idx, score in enumerate(scores)
          if score > confidence_threshold
      ]
      return predictions[keep]
  
imports: from extra.models.mask_rcnn import MaskRCNN
  from extra.models.resnet import ResNet
  from extra.models.mask_rcnn import BoxList
  from torch.nn import functional
  from torchvision import transforms
  from torchvision.transforms import functional
  import random
  from tinygrad.tensor import Tensor
  from PIL import Image
  import numpy
  import torch
  import argparse
  import cv2
label_desc: Use image node_id 4 for calling a global function with example usage: select_top_predictions(predictions, confidence_threshold) and returns: predictions