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 callback(self, pre, post, node_map):
      boxes = node_map[self.boxes][0]
      scores = node_map[self.scores][0]
      idxs = node_map[self.idxs][0]
      iou_thres = node_map[self.iou_threshold][0]
      num_boxes = node_map[self.num_boxes][0]
      indices = node_map[self.indices][0]
      return self.convert_batched_nms(
          boxes, scores, idxs, iou_thres, num_boxes, 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 3 for calling the MulticlassNMSRewrite obj's underlying member method code with example usage: obj.callback(pre, post, node_map) and returns: self