CommonUtility.h

// 常用工具

#ifndef __COMMON_UTILITY_H__
#define __COMMON_UTILITY_H__

#include <CommonDefinition.h>

namespace migraphxSamples
{

// 排序规则: 按照置信度或者按照面积排序
bool CompareConfidence(const ResultOfDetection& L, const ResultOfDetection& R);
bool CompareArea(const ResultOfDetection& L, const ResultOfDetection& R);

// 非极大抑制
void NMS(std::vector<ResultOfDetection>& detections, float IOUThreshold);

} // namespace migraphxSamples

#endif