Nms.cpp File Reference

#include "ml/contrib/Nms.h"
#include <Helper.h>
#include <tbb/parallel_for.h>
#include <iostream>
#include <numeric>
#include "ml/contrib/IoUImpl.h"

Include dependency graph for Nms.cpp:

Go to the source code of this file.

Namespaces
	cloudViewer
	Generic file read and write utility for python interface.
	cloudViewer::ml
	cloudViewer::ml::contrib

Functions
template<typename T >
static std::vector< int64_t >	cloudViewer::ml::contrib::SortIndexes (const T *values, int64_t num, bool descending=false)
static void	cloudViewer::ml::contrib::AllPairsSortedIoU (const float *boxes, const float *scores, const int64_t *sort_indices, uint64_t *mask, int n, double nms_overlap_thresh)
std::vector< int64_t >	cloudViewer::ml::contrib::NmsCPUKernel (const float *boxes, const float *scores, int n, double nms_overlap_thresh)