code: def decompress_masks(compressed_masks, num_queries):
      """Reconstructs the uncompressed mask representation."""
      vals, inds = compressed_masks
      b, _, h, w = vals.shape
      mask_probs = np.zeros((b, num_queries, h, w))
      ib, _, ih, iw = np.meshgrid(
          range(b), range(1), range(h), range(w), indexing="ij"
      )
      mask_probs[ib, inds, ih, iw] = vals
      return mask_probs
  
imports: from typing import Optional
  import jax
  import jax.numpy
  import numpy
  from PIL import Image
label_desc: Use image node_id 2 for calling a global function with example usage: decompress_masks(compressed_masks, num_queries) and returns: mask_probs