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.