"examples/dreambooth/train_dreambooth_flax.py" did not exist on "4b9f58952a5c9cbd7db71cf050dc7d3961505730"
CommonUtility.h 450 Bytes
Newer Older
Your Name's avatar
Your Name committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
// 常用工具

#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);

liucong's avatar
liucong committed
15
// 非极大抑制
Your Name's avatar
Your Name committed
16
17
18
19
20
void NMS(std::vector<ResultOfDetection> &detections, float IOUThreshold);

}

#endif