code: def count_pixels_in_batch(batch: BatchWrapper) -> int:
      """Count the pixels in the batch."""
      return sum(
          (
              image.shape[0] * image.shape[1]
              for image in batch.original_images
          )
      )
  
imports: import typing
  import numpy
  from deepchecks.core.check_result import CheckResult
  from deepchecks.core.checks import DatasetKind
  from deepchecks.core.condition import ConditionCategory
  from deepchecks.vision.base_checks import TrainTestCheck
  from deepchecks.vision.context import Context
  from deepchecks.vision.vision_data.batch_wrapper import BatchWrapper
  from deepchecks.vision.datasets.detection.coco_torch import load_dataset
  import pandas
  import plotly.express
  from deepchecks.core import ConditionResult
label_desc: Use image node_id 4 for calling a global function with example usage: count_pixels_in_batch(batch) and returns: sum