ArmNN
NotReleased
|
#include <reference/workloads/DetectionPostProcess.hpp>
#include <armnn/Descriptors.hpp>
#include <armnn/Types.hpp>
#include <boost/test/unit_test.hpp>
Go to the source code of this file.
Functions | |
BOOST_AUTO_TEST_CASE (TopKSortTest) | |
BOOST_AUTO_TEST_CASE (FullTopKSortTest) | |
BOOST_AUTO_TEST_CASE (IouTest) | |
BOOST_AUTO_TEST_CASE (NmsFunction) | |
void | DetectionPostProcessTestImpl (bool useRegularNms, const std::vector< float > &expectedDetectionBoxes, const std::vector< float > &expectedDetectionClasses, const std::vector< float > &expectedDetectionScores, const std::vector< float > &expectedNumDetections) |
BOOST_AUTO_TEST_CASE (RegularNmsDetectionPostProcess) | |
BOOST_AUTO_TEST_CASE (FastNmsDetectionPostProcess) | |
BOOST_AUTO_TEST_CASE | ( | TopKSortTest | ) |
Definition at line 15 of file RefDetectionPostProcessTests.cpp.
References armnn::TopKSort().
BOOST_AUTO_TEST_CASE | ( | FullTopKSortTest | ) |
Definition at line 26 of file RefDetectionPostProcessTests.cpp.
References armnn::TopKSort().
BOOST_AUTO_TEST_CASE | ( | IouTest | ) |
Definition at line 42 of file RefDetectionPostProcessTests.cpp.
References armnn::IntersectionOverUnion().
BOOST_AUTO_TEST_CASE | ( | NmsFunction | ) |
Definition at line 50 of file RefDetectionPostProcessTests.cpp.
References armnn::NonMaxSuppression(), and scores().
BOOST_AUTO_TEST_CASE | ( | RegularNmsDetectionPostProcess | ) |
Definition at line 173 of file RefDetectionPostProcessTests.cpp.
References DetectionPostProcessTestImpl().
BOOST_AUTO_TEST_CASE | ( | FastNmsDetectionPostProcess | ) |
Definition at line 189 of file RefDetectionPostProcessTests.cpp.
References BOOST_AUTO_TEST_SUITE_END(), and DetectionPostProcessTestImpl().
void DetectionPostProcessTestImpl | ( | bool | useRegularNms, |
const std::vector< float > & | expectedDetectionBoxes, | ||
const std::vector< float > & | expectedDetectionClasses, | ||
const std::vector< float > & | expectedDetectionScores, | ||
const std::vector< float > & | expectedNumDetections | ||
) |
Definition at line 72 of file RefDetectionPostProcessTests.cpp.
References anchors(), anchorsInfo, boxEncodings(), armnn::DetectionPostProcess(), armnn::Float32, DetectionPostProcessDescriptor::m_UseRegularNms, scores(), and scoresInfo.
Referenced by BOOST_AUTO_TEST_CASE().