class_docstr: WrongPredictionVis class for visualizing video inputs to Tensorboard
  for instances that the model makes wrong predictions.
class_docstr_tok: ["WrongPredictionVis","class","for","visualizing","video","inputs","to","Tensorboard","for","instances","that","the","model","makes","wrong","predictions","."]
docstr: Args:
      cfg (CfgNode): configs. Details can be found in
          slowfast/config/defaults.py
docstr_tok: ["Args",":","cfg","(","CfgNode",")",":","configs",".","Details","can","be","found","in","slowfast\/config\/defaults.py"]
code: def __init__(self, cfg):
      """
      Args:
          cfg (CfgNode): configs. Details can be found in
              slowfast/config/defaults.py
      """
      self.cfg = cfg
      self.class_names, _, self.subset = get_class_names(
          cfg.TENSORBOARD.CLASS_NAMES_PATH,
          subset_path=cfg.TENSORBOARD.WRONG_PRED_VIS.SUBSET_PATH,
      )
      if self.subset is not None:
          self.subset = set(self.subset)
      self.num_class = cfg.MODEL.NUM_CLASSES
      self.video_vis = VideoVisualizer(
          cfg.MODEL.NUM_CLASSES,
          cfg.TENSORBOARD.CLASS_NAMES_PATH,
          1,
          cfg.TENSORBOARD.MODEL_VIS.COLORMAP,
      )
      self.tag = cfg.TENSORBOARD.WRONG_PRED_VIS.TAG
      self.writer = tb.TensorboardWriter(cfg)
      self.model_incorrect_classes = set()
  
file_path: SlowFast/slowfast/visualization/prediction_vis.py
imports: import numpy
  import torch
  import slowfast.datasets.utils
  import slowfast.utils.logging
  import slowfast.visualization.tensorboard_vis
  from slowfast.utils.misc import get_class_names
  from slowfast.visualization.video_visualizer import VideoVisualizer
label_desc: Use image node_id 1 to create a new WrongPredictionVis object  with example: obj = WrongPredictionVis(cfg)