7 #include <boost/algorithm/string.hpp> 8 #include <boost/test/unit_test.hpp> 10 #include <boost/filesystem.hpp> 11 #include <boost/optional.hpp> 12 #include <boost/variant.hpp> 21 const std::map<std::string, std::string> GetValidationLabelSet()
23 std::map<std::string, std::string> validationLabelSet;
24 validationLabelSet.insert(std::make_pair(
"val_01.JPEG",
"goldfinch"));
25 validationLabelSet.insert(std::make_pair(
"val_02.JPEG",
"magpie"));
26 validationLabelSet.insert(std::make_pair(
"val_03.JPEG",
"brambling"));
27 validationLabelSet.insert(std::make_pair(
"val_04.JPEG",
"robin"));
28 validationLabelSet.insert(std::make_pair(
"val_05.JPEG",
"indigo bird"));
29 validationLabelSet.insert(std::make_pair(
"val_06.JPEG",
"ostrich"));
30 validationLabelSet.insert(std::make_pair(
"val_07.JPEG",
"jay"));
31 validationLabelSet.insert(std::make_pair(
"val_08.JPEG",
"snowbird"));
32 validationLabelSet.insert(std::make_pair(
"val_09.JPEG",
"house finch"));
33 validationLabelSet.insert(std::make_pair(
"val_09.JPEG",
"bulbul"));
35 return validationLabelSet;
37 const std::vector<armnnUtils::LabelCategoryNames> GetModelOutputLabels()
39 const std::vector<armnnUtils::LabelCategoryNames> modelOutputLabels =
41 {
"ostrich",
"Struthio camelus"},
42 {
"brambling",
"Fringilla montifringilla"},
43 {
"goldfinch",
"Carduelis carduelis"},
44 {
"house finch",
"linnet",
"Carpodacus mexicanus"},
45 {
"junco",
"snowbird"},
46 {
"indigo bunting",
"indigo finch",
"indigo bird",
"Passerina cyanea"},
47 {
"robin",
"American robin",
"Turdus migratorius"},
52 return modelOutputLabels;
66 std::vector<float> inferenceOutputVector1 = {0.05f, 0.10f, 0.70f, 0.15f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f};
67 TContainer inference1Container(inferenceOutputVector1);
68 std::vector<TContainer> outputTensor1;
69 outputTensor1.push_back(inference1Container);
71 std::string imageName =
"val_01.JPEG";
79 std::vector<float> inferenceOutputVector2 = {0.10f, 0.0f, 0.0f, 0.0f, 0.05f, 0.70f, 0.0f, 0.0f, 0.0f, 0.15f};
80 TContainer inference2Container(inferenceOutputVector2);
81 std::vector<TContainer> outputTensor2;
82 outputTensor2.push_back(inference2Container);
84 imageName =
"val_02.JPEG";
96 std::vector<float> inferenceOutputVector3 = {0.0f, 0.10f, 0.0f, 0.0f, 0.05f, 0.70f, 0.0f, 0.0f, 0.0f, 0.15f};
97 TContainer inference3Container(inferenceOutputVector3);
98 std::vector<TContainer> outputTensor3;
99 outputTensor3.push_back(inference3Container);
101 imageName =
"val_03.JPEG";
float GetAccuracy(unsigned int k)
void AddImageResult(const std::string &imageName, std::vector< TContainer > outputTensor)
BOOST_CHECK(profilingService.GetCurrentState()==ProfilingState::WaitingForAck)
BOOST_AUTO_TEST_SUITE_END()
boost::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > > TContainer
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(TestFloat32OutputTensorAccuracy, TestHelper)