Lines Matching refs:select
360 std::function<const float*(uint32_t)> getRoiBase, uint32_t* select, in hardNmsSingleClass() argument
362 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in hardNmsSingleClass()
391 std::vector<uint32_t>* select) { in hardNmsMultiClass() argument
394 uint32_t size = select->size(); in hardNmsMultiClass()
399 select->push_back(index); in hardNmsMultiClass()
402 uint32_t* selectStart = select->data() + size; in hardNmsMultiClass()
403 uint32_t selectLength = select->size() - size; in hardNmsMultiClass()
406 select->resize(selectEnd - select->data()); in hardNmsMultiClass()
410 std::sort(select->begin(), select->end(), in hardNmsMultiClass()
414 if (maxNumDetections < 0 || select->size() <= maxNumDetections) { in hardNmsMultiClass()
417 select->resize(maxNumDetections); in hardNmsMultiClass()
424 uint32_t* select, uint32_t selectLength) { in softNmsSingleClass() argument
425 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in softNmsSingleClass()
455 std::vector<uint32_t>* select) { in softNmsMultiClass() argument
458 uint32_t size = select->size(); in softNmsMultiClass()
463 select->push_back(index); in softNmsMultiClass()
466 uint32_t* selectStart = select->data() + size; in softNmsMultiClass()
467 uint32_t selectLength = select->size() - size; in softNmsMultiClass()
471 select->resize(selectEnd - select->data()); in softNmsMultiClass()
475 std::sort(select->begin(), select->end(), in softNmsMultiClass()
479 if (maxNumDetections < 0 || select->size() <= maxNumDetections) { in softNmsMultiClass()
482 select->resize(maxNumDetections); in softNmsMultiClass()
934 std::vector<uint32_t>* select) { in filterBoxes() argument
937 for (uint32_t j = 0; j < select->size(); j++) { in filterBoxes()
938 const float* roiInfo = roiBase + (*select)[j] * kRoiDim; in filterBoxes()
946 (*select)[i++] = (*select)[j]; in filterBoxes()
949 select->resize(i); in filterBoxes()
1017 std::vector<uint32_t> select(batchSize); in generateProposalsNhwcFloat32Compute() local
1018 std::iota(select.begin(), select.end(), 0); in generateProposalsNhwcFloat32Compute()
1019 if (preNmsTopN > 0 && preNmsTopN < select.size()) { in generateProposalsNhwcFloat32Compute()
1020 std::sort(select.begin(), select.end(), in generateProposalsNhwcFloat32Compute()
1024 select.resize(preNmsTopN); in generateProposalsNhwcFloat32Compute()
1028 filterBoxes(roiTransformedBuffer.data(), imageInfoBase, minSize, &select); in generateProposalsNhwcFloat32Compute()
1036 select.data(), select.size()); in generateProposalsNhwcFloat32Compute()
1037 uint32_t selectSize = selectEnd - select.data(); in generateProposalsNhwcFloat32Compute()
1038 select.resize(selectSize); in generateProposalsNhwcFloat32Compute()
1041 for (auto i : select) { in generateProposalsNhwcFloat32Compute()
1483 std::vector<uint32_t> select; in detectionPostprocessFloat32() local
1490 &select); in detectionPostprocessFloat32()
1491 for (uint32_t i = 0; i < select.size(); i++) { in detectionPostprocessFloat32()
1492 uint32_t ind = select[i]; in detectionPostprocessFloat32()
1498 *detectionOutData++ = select.size(); in detectionPostprocessFloat32()
1506 std::vector<uint32_t> select; in detectionPostprocessFloat32() local
1509 select.push_back(a); in detectionPostprocessFloat32()
1515 select.data(), select.size()); in detectionPostprocessFloat32()
1516 select.resize(selectEnd - select.data()); in detectionPostprocessFloat32()
1520 for (auto i : select) { in detectionPostprocessFloat32()
1535 *detectionOutData++ = select.size() * numOutClasses; in detectionPostprocessFloat32()