class_docstr: A callback to rewrite nms and restore batched nms.
class_docstr_tok: ["A","callback","to","rewrite","nms","and","restore","batched","nms","."]
code: def __init__(self):
      super().__init__()
      # exprs to extract
      self.boxes = wildcard()
      self.scores = wildcard()
      self.idxs = wildcard()
      self.iou_threshold = wildcard()
      self.num_boxes = wildcard()
      self.indices = wildcard()
  
      self.pattern = batched_nms_pattern(
          self.boxes,
          self.scores,
          self.idxs,
          self.iou_threshold,
          self.num_boxes,
          self.indices,
      )
  
imports: from ..None import expr
  from ..None import op
  from ..dataflow_pattern import wildcard, is_constant, is_op, rewrite, is_tuple, is_tuple_get_item, is_if, DFPatternCallback
label_desc: Use image node_id 1 to create a new MulticlassNMSRewrite object from inherited base classes: DFPatternCallback with example: obj = MulticlassNMSRewrite()