non_maximal_supression.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_METRICS_NMS_HPP
#define MLPACK_CORE_METRICS_NMS_HPP

namespace mlpack {
namespace metric {

template<bool UseCoordinates = false>
class NMS
{
 public:
  NMS() { /* Nothing to do here. */ }

  template<
      typename BoundingBoxesType,
      typename ConfidenceScoreType,
      typename OutputType
  >
  static void Evaluate(const BoundingBoxesType& boundingBoxes,
                       const ConfidenceScoreType& confidenceScores,
                       OutputType& selectedIndices,
                       const double threshold = 0.5);

  static const bool useCoordinates = UseCoordinates;

  template <typename Archive>
  void serialize(Archive &ar, const uint32_t /* version */);
}; // Class NMS.

} // namespace metric
} // namespace mlpack

// Include implementation.
#include "non_maximal_supression_impl.hpp"

#endif