code: def _get_predicted_classes_per_image_semantic_segmentation(
      predictions: List[np.ndarray],
  ) -> List[List[int]]:
      """Return a list containing the classes in batch."""
      return [
          np.unique(pred.argmax(0)).tolist() for pred in predictions
      ]
  
file_path: deepchecks/deepchecks/vision/utils/label_prediction_properties.py
imports: from typing import List, Sequence, Union
  import numpy
label_desc: Use image node_id 11 for calling a global function with example usage: _get_predicted_classes_per_image_semantic_segmentation(predictions) and returns: unknown