Searched refs:iouThreshold (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 359 uint32_t* hardNmsSingleClass(const float* scoresData, float iouThreshold, int32_t maxNumDetections, in hardNmsSingleClass() argument
377 if (iou >= iouThreshold) { in hardNmsSingleClass()
388 float scoreThreshold, float iouThreshold, int32_t maxNumDetections, in hardNmsMultiClass() argument
404 uint32_t* selectEnd = hardNmsSingleClass(scoresData, iouThreshold, maxNumDetectionsPerClass, in hardNmsMultiClass()
489 int32_t softNmsKernel, float iouThreshold, float sigma, in boxWithNmsLimitFloat32Compute() argument
495 kernel = [&iouThreshold](float iou) { return iou < iouThreshold ? 1.0f : 0.0f; }; in boxWithNmsLimitFloat32Compute()
497 kernel = [&iouThreshold](float iou) { return iou < iouThreshold ? 1.0f : 1.0f - iou; }; in boxWithNmsLimitFloat32Compute()
620 int32_t maxNumDetections, int32_t softNmsKernel, float iouThreshold, in boxWithNmsLimitFloat32() argument
633 scoreThreshold, maxNumDetections, softNmsKernel, iouThreshold, sigma, nmsScoreThreshold, in boxWithNmsLimitFloat32()
643 int32_t softNmsKernel, _Float16 iouThreshold, _Float16 sigma, in boxWithNmsLimitFloat16() argument
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3511 ANeuralNetworksOperandType iouThreshold = scaleScalar; in detectionPostprocessingOpTest() local
3527 numOfDetections, scoreThreshold, iouThreshold, includeBackground}, in detectionPostprocessingOpTest()
|