code: def setup_dataset(dataset_name):
      logger.info("Loading dataset {}".format(dataset_name))
      start = timer()
      dataset = DatasetCatalog.get(dataset_name)
      stop = timer()
      logger.info(
          "Loaded dataset {} in {:.3f}s".format(
              dataset_name, stop - start
          )
      )
      return dataset
  
imports: import argparse
  import logging
  import os
  import sys
  from timeit import default_timer
  from typing import Any, ClassVar, Dict, List
  import torch
  from detectron2.data.catalog import DatasetCatalog
  from detectron2.utils.file_io import PathManager
  from detectron2.utils.logger import setup_logger
  from densepose.structures import DensePoseDataRelative
  from densepose.utils.dbhelper import EntrySelector
  from densepose.utils.logger import verbosity_to_level
  from densepose.vis.base import CompoundVisualizer
  from densepose.vis.bounding_box import BoundingBoxVisualizer
  from densepose.vis.densepose_data_points import DensePoseDataCoarseSegmentationVisualizer, DensePoseDataPointsIVisualizer, DensePoseDataPointsUVisualizer, DensePoseDataPointsVisualizer, DensePoseDataPointsVVisualizer
label_desc: Use image node_id 2 for calling a global function with example usage: setup_dataset(dataset_name) and returns: dataset