diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
commit | c577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch) | |
tree | bd7d4c148df27f8be6649d313efb24f536b7cf34 /tests/YoloDatabase.cpp | |
parent | 4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff) | |
download | armnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz |
Release 18.08
Diffstat (limited to 'tests/YoloDatabase.cpp')
-rw-r--r-- | tests/YoloDatabase.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/YoloDatabase.cpp b/tests/YoloDatabase.cpp index 4c91384073..71362b2218 100644 --- a/tests/YoloDatabase.cpp +++ b/tests/YoloDatabase.cpp @@ -78,12 +78,12 @@ std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsig const auto& testCaseInputOutput = g_PerTestCaseInputOutput[testCaseId]; const std::string imagePath = m_ImageDir + testCaseInputOutput.first; - // Load test case input image + // Loads test case input image. std::vector<float> imageData; try { InferenceTestImage image(imagePath.c_str()); - image.Resize(YoloImageWidth, YoloImageHeight); + image.Resize(YoloImageWidth, YoloImageHeight, CHECK_LOCATION()); imageData = GetImageDataInArmNnLayoutAsNormalizedFloats(ImageChannelLayout::Rgb, image); } catch (const InferenceTestImageException& e) @@ -92,10 +92,10 @@ std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsig return nullptr; } - // Prepare test case output + // Prepares test case output. std::vector<YoloDetectedObject> topObjectDetections; topObjectDetections.reserve(1); topObjectDetections.push_back(testCaseInputOutput.second); return std::make_unique<YoloTestCaseData>(std::move(imageData), std::move(topObjectDetections)); -}
\ No newline at end of file +} |