Lines Matching refs:select
363 std::function<const float*(uint32_t)> getRoiBase, uint32_t* select, in hardNmsSingleClass() argument
365 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in hardNmsSingleClass()
394 std::vector<uint32_t>* select) { in hardNmsMultiClass() argument
397 uint32_t size = select->size(); in hardNmsMultiClass()
402 select->push_back(index); in hardNmsMultiClass()
405 uint32_t* selectStart = select->data() + size; in hardNmsMultiClass()
406 uint32_t selectLength = select->size() - size; in hardNmsMultiClass()
409 select->resize(selectEnd - select->data()); in hardNmsMultiClass()
413 std::sort(select->begin(), select->end(), in hardNmsMultiClass()
417 if (maxNumDetections < 0 || select->size() <= maxNumDetections) { in hardNmsMultiClass()
420 select->resize(maxNumDetections); in hardNmsMultiClass()
427 uint32_t* select, uint32_t selectLength) { in softNmsSingleClass() argument
428 uint32_t *selectStart = select, *selectEnd = select + selectLength, numDetections = 0; in softNmsSingleClass()
458 std::vector<uint32_t>* select) { in softNmsMultiClass() argument
461 uint32_t size = select->size(); in softNmsMultiClass()
466 select->push_back(index); in softNmsMultiClass()
469 uint32_t* selectStart = select->data() + size; in softNmsMultiClass()
470 uint32_t selectLength = select->size() - size; in softNmsMultiClass()
474 select->resize(selectEnd - select->data()); in softNmsMultiClass()
478 std::sort(select->begin(), select->end(), in softNmsMultiClass()
482 if (maxNumDetections < 0 || select->size() <= maxNumDetections) { in softNmsMultiClass()
485 select->resize(maxNumDetections); in softNmsMultiClass()
941 std::vector<uint32_t>* select) { in filterBoxes() argument
944 for (uint32_t j = 0; j < select->size(); j++) { in filterBoxes()
945 const float* roiInfo = roiBase + (*select)[j] * kRoiDim; in filterBoxes()
953 (*select)[i++] = (*select)[j]; in filterBoxes()
956 select->resize(i); in filterBoxes()
1024 std::vector<uint32_t> select(batchSize); in generateProposalsNhwcFloat32Compute() local
1025 std::iota(select.begin(), select.end(), 0); in generateProposalsNhwcFloat32Compute()
1026 if (preNmsTopN > 0 && preNmsTopN < select.size()) { in generateProposalsNhwcFloat32Compute()
1027 std::sort(select.begin(), select.end(), in generateProposalsNhwcFloat32Compute()
1031 select.resize(preNmsTopN); in generateProposalsNhwcFloat32Compute()
1035 filterBoxes(roiTransformedBuffer.data(), imageInfoBase, minSize, &select); in generateProposalsNhwcFloat32Compute()
1043 select.data(), select.size()); in generateProposalsNhwcFloat32Compute()
1044 uint32_t selectSize = selectEnd - select.data(); in generateProposalsNhwcFloat32Compute()
1045 select.resize(selectSize); in generateProposalsNhwcFloat32Compute()
1048 for (auto i : select) { in generateProposalsNhwcFloat32Compute()
1494 std::vector<uint32_t> select; in detectionPostprocessFloat32() local
1501 &select); in detectionPostprocessFloat32()
1502 for (uint32_t i = 0; i < select.size(); i++) { in detectionPostprocessFloat32()
1503 uint32_t ind = select[i]; in detectionPostprocessFloat32()
1509 *detectionOutData++ = select.size(); in detectionPostprocessFloat32()
1517 std::vector<uint32_t> select; in detectionPostprocessFloat32() local
1520 select.push_back(a); in detectionPostprocessFloat32()
1526 select.data(), select.size()); in detectionPostprocessFloat32()
1527 select.resize(selectEnd - select.data()); in detectionPostprocessFloat32()
1531 for (auto i : select) { in detectionPostprocessFloat32()
1546 *detectionOutData++ = select.size() * numOutClasses; in detectionPostprocessFloat32()