code: def init_pixel_counts_dict() -> t.Dict[str, int]:
      """Initialize the pixel counts dicts."""
      return {
          DatasetKind.TRAIN.value: 0,
          DatasetKind.TEST.value: 0,
      }
  
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 2 for calling a global function with example usage: init_pixel_counts_dict() and returns: dict