code: def vis(dataset, detections_pkl, thresh, output_dir, limit=0):
      ds = JsonDataset(dataset)
      roidb = ds.get_roidb()
  
      with open(detections_pkl, "r") as f:
          dets = pickle.load(f)
  
      assert all(
          k in dets for k in ["all_boxes", "all_segms", "all_keyps"]
      ), "Expected detections pkl file in the format used by test_engine.py"
  
      all_boxes = dets["all_boxes"]
      all_segms = dets["all_segms"]
      all_keyps = dets["all_keyps"]
  
      def id_or_index(ix, val):
          if len(val) == 0:
              return val
          else:
              return val[ix]
  
      for ix, entry in enumerate(roidb):
          if limit > 0 and ix >= limit:
              break
          if ix % 10 == 0:
              print("{:d}/{:d}".format(ix + 1, len(roidb)))
  
          im = cv2.imread(entry["image"])
          im_name = os.path.splitext(os.path.basename(entry["image"]))[
              0
          ]
  
          cls_boxes_i = [
              id_or_index(ix, cls_k_boxes) for cls_k_boxes in all_boxes
          ]
          cls_segms_i = [
              id_or_index(ix, cls_k_segms) for cls_k_segms in all_segms
          ]
          cls_keyps_i = [
              id_or_index(ix, cls_k_keyps) for cls_k_keyps in all_keyps
          ]
  
          vis_utils.vis_one_image(
              im[:, :, ::-1],
              "{:d}_{:s}".format(ix, im_name),
              os.path.join(output_dir, "vis"),
              cls_boxes_i,
              segms=cls_segms_i,
              keypoints=cls_keyps_i,
              thresh=thresh,
              box_alpha=0.8,
              dataset=ds,
              show_class=True,
          )
  
file_path: dense-pose/tools/visualize_results.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from __future__ import unicode_literals
  import argparse
  import cPickle
  import cv2
  import os
  import sys
  from detectron.datasets.json_dataset import JsonDataset
  import detectron.utils.vis
label_desc: Use image node_id 2 for calling a global function with example usage: vis(dataset, detections_pkl, thresh, output_dir, limit) and returns: val, val