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 unsigned int /* version */);
}; // Class NMS.
 
} // namespace metric
} // namespace mlpack
 
// Include implementation.
#include "non_maximal_supression_impl.hpp"
 
#endif