Function mindspore::dataset::ApplyNms

Function Documentation

std::vector<int> mindspore::dataset::ApplyNms(const std::vector<std::vector<float>> &all_boxes, std::vector<float> &all_scores, float thres, int max_boxes)

Apply Non-Maximum Suppression.

Parameters
  • all_boxes[in] All input boxes.

  • all_scores[in] Score after all boxes are executed through the network.

  • thres[in] Pre-value of IOU.

  • max_boxes[in] Maximum value of output box.